Searched refs:TensorMemoryType (Results 1 – 4 of 4) sorted by relevance
472 InferenceContext::TensorMemoryType InferenceContext::GetTensorMemoryType( in GetTensorMemoryType()475 return TensorMemoryType::kConst; in GetTensorMemoryType()477 return TensorMemoryType::kVariable; in GetTensorMemoryType()479 return TensorMemoryType::kBuffer; in GetTensorMemoryType()481 return TensorMemoryType::kStrongShape; in GetTensorMemoryType()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()
105 enum class TensorMemoryType { kStrongShape, kBuffer, kVariable, kConst }; enum134 TensorMemoryType GetTensorMemoryType(ValueId id);
429 InferenceContext::TensorMemoryType InferenceContext::GetTensorMemoryType( in GetTensorMemoryType()432 return TensorMemoryType::kPreallocated; in GetTensorMemoryType()434 return TensorMemoryType::kConst; in GetTensorMemoryType()436 return TensorMemoryType::kBuffer; in GetTensorMemoryType()438 return TensorMemoryType::kStrongShape; in GetTensorMemoryType()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()
121 enum class TensorMemoryType { enum148 TensorMemoryType GetTensorMemoryType(ValueId id);