| 246 | } |
| 247 | |
| 248 | int3 DepthWiseConvolution::GetGridSize() const { |
| 249 | const int grid_x = dst_[0]->Width(); |
| 250 | const int grid_y = dst_[0]->Height(); |
| 251 | const int grid_z = dst_[0]->Depth(); |
| 252 | return int3(grid_x, grid_y, grid_z); |
| 253 | } |
| 254 | |
| 255 | Status DepthWiseConvolution::Tune(const TuningParameters& params) { |
| 256 | RETURN_IF_ERROR(BindArguments()); |