Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dmetal_arguments.h96 absl::Status SetImage3D(const std::string& name, id<MTLTexture> handle);
Dmetal_arguments.cc553 RETURN_IF_ERROR(SetImage3D(absl::StrCat(name, "_", r.first), r.second)); in SetGPUResources()
645 absl::Status MetalArguments::SetImage3D(const std::string& name, in SetImage3D() function in tflite::gpu::metal::MetalArguments
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dcl_arguments.h105 absl::Status SetImage3D(const std::string& name, cl_mem memory);
Dcl_arguments.cc618 absl::Status CLArguments::SetImage3D(const std::string& name, cl_mem memory) { in SetImage3D() function in tflite::gpu::cl::CLArguments
681 RETURN_IF_ERROR(SetImage3D(absl::StrCat(name, "_", r.first), r.second)); in SetGPUResources()