Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dmetal_arguments.h97 absl::Status SetImageBuffer(const std::string& name, id<MTLTexture> handle);
Dmetal_arguments.cc556 RETURN_IF_ERROR(SetImageBuffer(absl::StrCat(name, "_", r.first), r.second)); in SetGPUResources()
656 absl::Status MetalArguments::SetImageBuffer(const std::string& name, in SetImageBuffer() function in tflite::gpu::metal::MetalArguments
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_arguments.h106 absl::Status SetImageBuffer(const std::string& name, cl_mem memory);
Dcl_arguments.cc628 absl::Status CLArguments::SetImageBuffer(const std::string& name, in SetImageBuffer() function in tflite::gpu::cl::CLArguments
684 RETURN_IF_ERROR(SetImageBuffer(absl::StrCat(name, "_", r.first), r.second)); in SetGPUResources()