Searched refs:GetTensorMemoryType (Results 1 – 2 of 2) sorted by relevance
398 InferenceContext::TensorMemoryType InferenceContext::GetTensorMemoryType( in GetTensorMemoryType() function in tflite::gpu::metal::InferenceContext455 return GetTensorMemoryType(id) == TensorMemoryType::kBuffer; in AllocateMemoryForBuffers()562 if (GetTensorMemoryType(tensor_id) != TensorMemoryType::kBuffer) { in AllocateMemoryForBuffers()603 return GetTensorMemoryType(id) == TensorMemoryType::kStrongShape; in AllocateMemoryForStrongShapes()634 if (GetTensorMemoryType(tensor_id) != TensorMemoryType::kStrongShape) { in AllocateMemoryForStrongShapes()
164 TensorMemoryType GetTensorMemoryType(ValueId id);