Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.h141 absl::Status AllocateMemoryForBuffers(MetalDevice* device);
Dinference_context.cc379 RETURN_IF_ERROR(AllocateMemoryForBuffers(device)); in AllocateTensors()
478 absl::Status InferenceContext::AllocateMemoryForBuffers(MetalDevice* device) { in AllocateMemoryForBuffers() function in tflite::gpu::metal::InferenceContext
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.h126 absl::Status AllocateMemoryForBuffers(CLContext* context);
Dinference_context.cc488 RETURN_IF_ERROR(AllocateMemoryForBuffers(context)); in AllocateMemory()
521 absl::Status InferenceContext::AllocateMemoryForBuffers(CLContext* context) { in AllocateMemoryForBuffers() function in tflite::gpu::cl::InferenceContext