Method GetGridSize

tensorflow/lite/delegates/gpu/cl/kernels/conv_buffer_1x1.cc:278–292

}

int3 ConvBuffer1x1::GetGridSize() const {
  if (src_[0]->Width() % 2 == 0) {  // using kernel_flt8_
    const int grid_x =
        IntegralDivideRoundUp(GetGridWidth(dst_[0]->Width()), flt8_x_count_);
    const int grid_y = IntegralDivideRoundUp(dst_[0]->Height(), flt8_y_count_);
    const int grid_z = dst_[0]->Depth();
    return int3(grid_x, grid_y, grid_z);
  } else {  // using kernel_flt4_
    const int grid_x =
        IntegralDivideRoundUp(GetGridWidth(dst_[0]->Width()), flt4_x_count_);
    const int grid_y = IntegralDivideRoundUp(dst_[0]->Height(), flt4_y_count_);
    const int grid_z = dst_[0]->Depth();
    return int3(grid_x, grid_y, grid_z);
  }
}

Status ConvBuffer1x1::Tune(const TuningParameters& params) {
  RETURN_IF_ERROR(BindArguments());