Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.cc438 return TensorMemoryType::kStrongShape; in GetTensorMemoryType()
549 return GetTensorMemoryType(id) == TensorMemoryType::kStrongShape; in AllocateMemoryForStrongShapes()
571 if (GetTensorMemoryType(tensor_id) != TensorMemoryType::kStrongShape) { in AllocateMemoryForStrongShapes()
Dinference_context.h122 kStrongShape, enumerator
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.cc481 return TensorMemoryType::kStrongShape; in GetTensorMemoryType()
579 return GetTensorMemoryType(id) == TensorMemoryType::kStrongShape; in AllocateMemoryForStrongShapes()
599 if (GetTensorMemoryType(t.first) != TensorMemoryType::kStrongShape) { in AllocateMemoryForStrongShapes()
Dinference_context.h105 enum class TensorMemoryType { kStrongShape, kBuffer, kVariable, kConst }; enumerator