Method Compile

tensorflow/lite/delegates/gpu/cl/kernels/conv_constants.cc:221–234

}

Status ConvConstants::Compile(const CreationContext& creation_context) {
  const auto code = GenerateConvolutionConstantCode(
      definition_.src_tensors[0], definition_.dst_tensors[0],
      definition_.precision, kernel_size_, dilation_, src_channels_,
      dst_channels_, *creation_context.device, linked_operations_);
  std::vector<CompilerOptions> options;
  if (definition_.precision == CalculationsPrecision::F16 &&
      creation_context.device->IsAdreno3xx()) {
    options.push_back(CompilerOptions::ADRENO_FULL_SIMD_LINE);
  }
  return creation_context.cache->GetOrCreateCLKernel(
      code, "main_function", options, *creation_context.context,
      *creation_context.device, &kernel_);
}

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