Method GetGridSize

tensorflow/lite/delegates/gpu/cl/kernels/softmax.cc:113–118

}

int3 Softmax::GetGridSize() const {
  const int grid_x = dst_[0]->Width();
  const int grid_y = dst_[0]->Height();
  const int grid_z = 1;
  return int3(grid_x, grid_y, grid_z);
}

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