Searched refs:SetCustomMemory (Results 1 – 2 of 2) sorted by relevance
107 absl::Status SetCustomMemory(const std::string& name, cl_mem memory);
639 absl::Status CLArguments::SetCustomMemory(const std::string& name, in SetCustomMemory() function in tflite::gpu::cl::CLArguments688 SetCustomMemory(absl::StrCat(name, "_", r.first), r.second)); in SetGPUResources()