Method Compile

tensorflow/lite/delegates/gpu/cl/kernels/softmax.cc:93–100

}

Status Softmax::Compile(const CreationContext& creation_context) {
  const auto code = GetSoftmaxKernelCode(
      definition_.src_tensors[0], definition_.dst_tensors[0],
      definition_.precision, linked_operations_);
  return creation_context.cache->GetOrCreateCLKernel(
      code, "main_function", *creation_context.context,
      *creation_context.device, &kernel_);
}

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