Method Compile

tensorflow/lite/delegates/gpu/cl/kernels/pooling.cc:195–219

}

Status Pooling::Compile(const CreationContext& creation_context) {
  std::string code;
  const bool manual_boundary_check =
      definition_.src_tensors[0].storage_type == TensorStorageType::BUFFER ||
      creation_context.device->IsAdreno3xx();
  switch (type_) {
    case PoolingType::AVERAGE:
      code = GetAveragePoolingKernelCode(
          definition_.src_tensors[0], definition_.dst_tensors[0],
          definition_.precision, linked_operations_, manual_boundary_check);
      break;
    case PoolingType::MAX:
      code = GetMaxPoolingKernelCode(
          definition_.src_tensors[0], definition_.dst_tensors[0],
          definition_.precision, linked_operations_, output_indices_);
      break;
    default:
      return InvalidArgumentError(
          "You should create another kernel with this params");
      break;
  }
  return creation_context.cache->GetOrCreateCLKernel(
      code, "main_function", *creation_context.context,
      *creation_context.device, &kernel_);
}

Status Pooling::BindArguments() {
  kernel_.ResetBindingCounter();