MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / Compile

Method Compile

tensorflow/lite/delegates/gpu/cl/kernels/max_unpooling.cc:120–131  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

118}
119
120Status MaxUnpooling::Compile(const CreationContext& creation_context) {
121 const bool manual_boundary_check =
122 definition_.src_tensors[0].storage_type == TensorStorageType::BUFFER ||
123 creation_context.device->IsAdreno3xx();
124 const auto code = GetMaxUnoolingKernelCode(
125 definition_.src_tensors[0], definition_.src_tensors[1],
126 definition_.dst_tensors[0], definition_.precision, linked_operations_,
127 manual_boundary_check);
128 return creation_context.cache->GetOrCreateCLKernel(
129 code, "main_function", *creation_context.context,
130 *creation_context.device, &kernel_);
131}
132
133Status MaxUnpooling::BindArguments() {
134 kernel_.ResetBindingCounter();

Callers

nothing calls this directly

Calls 3

GetMaxUnoolingKernelCodeFunction · 0.85
IsAdreno3xxMethod · 0.80
GetOrCreateCLKernelMethod · 0.80

Tested by

no test coverage detected