Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.h142 absl::Status AllocateMemoryForStrongShapes(MetalDevice* device);
Dinference_context.cc380 RETURN_IF_ERROR(AllocateMemoryForStrongShapes(device)); in AllocateTensors()
544 absl::Status InferenceContext::AllocateMemoryForStrongShapes( in AllocateMemoryForStrongShapes() function in tflite::gpu::metal::InferenceContext
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.h128 absl::Status AllocateMemoryForStrongShapes(CLContext* context);
Dinference_context.cc489 RETURN_IF_ERROR(AllocateMemoryForStrongShapes(context)); in AllocateMemory()
574 absl::Status InferenceContext::AllocateMemoryForStrongShapes( in AllocateMemoryForStrongShapes() function in tflite::gpu::cl::InferenceContext