Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.cc472 InferenceContext::TensorMemoryType InferenceContext::GetTensorMemoryType( in GetTensorMemoryType() function in tflite::gpu::cl::InferenceContext
525 return GetTensorMemoryType(id) == TensorMemoryType::kBuffer; in AllocateMemoryForBuffers()
560 if (GetTensorMemoryType(t.first) != TensorMemoryType::kBuffer) continue; in AllocateMemoryForBuffers()
579 return GetTensorMemoryType(id) == TensorMemoryType::kStrongShape; in AllocateMemoryForStrongShapes()
599 if (GetTensorMemoryType(t.first) != TensorMemoryType::kStrongShape) { in AllocateMemoryForStrongShapes()
Dinference_context.h134 TensorMemoryType GetTensorMemoryType(ValueId id);
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.cc429 InferenceContext::TensorMemoryType InferenceContext::GetTensorMemoryType( in GetTensorMemoryType() function in tflite::gpu::metal::InferenceContext
482 return GetTensorMemoryType(id) == TensorMemoryType::kBuffer; in AllocateMemoryForBuffers()
549 return GetTensorMemoryType(id) == TensorMemoryType::kStrongShape; in AllocateMemoryForStrongShapes()
571 if (GetTensorMemoryType(tensor_id) != TensorMemoryType::kStrongShape) { in AllocateMemoryForStrongShapes()
Dinference_context.h148 TensorMemoryType GetTensorMemoryType(ValueId id);