Home
last modified time | relevance | path

Searched refs:SetImage2D (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dmetal_arguments.h94 absl::Status SetImage2D(const std::string& name, id<MTLTexture> handle);
Dmetal_arguments.cc546 RETURN_IF_ERROR(SetImage2D(absl::StrCat(name, "_", r.first), r.second)); in SetGPUResources()
623 absl::Status MetalArguments::SetImage2D(const std::string& name, in SetImage2D() function in tflite::gpu::metal::MetalArguments
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_arguments.h102 absl::Status SetImage2D(const std::string& name, cl_mem memory);
Dcl_arguments.cc587 absl::Status CLArguments::SetImage2D(const std::string& name, cl_mem memory) { in SetImage2D() function in tflite::gpu::cl::CLArguments
674 RETURN_IF_ERROR(SetImage2D(absl::StrCat(name, "_", r.first), r.second)); in SetGPUResources()