Searched refs:TensorMemoryType (Results 1 – 4 of 4) sorted by relevance
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()529 if (GetTensorMemoryType(tensor_id) != TensorMemoryType::kBuffer) { in AllocateMemoryForBuffers()550 return GetTensorMemoryType(id) == TensorMemoryType::kStrongShape; in AllocateMemoryForStrongShapes()572 if (GetTensorMemoryType(tensor_id) != TensorMemoryType::kStrongShape) { in AllocateMemoryForStrongShapes()
121 enum class TensorMemoryType { enum149 TensorMemoryType GetTensorMemoryType(ValueId id);
504 InferenceContext::TensorMemoryType InferenceContext::GetTensorMemoryType( in GetTensorMemoryType()507 return TensorMemoryType::kConst; in GetTensorMemoryType()509 return TensorMemoryType::kVariable; in GetTensorMemoryType()512 return TensorMemoryType::kBuffer; in GetTensorMemoryType()514 return TensorMemoryType::kStrongShape; in GetTensorMemoryType()560 return GetTensorMemoryType(gpu_info, id) == TensorMemoryType::kBuffer; in AllocateMemoryForBuffers()642 if (GetTensorMemoryType(gpu_info, t.first) != TensorMemoryType::kBuffer) in AllocateMemoryForBuffers()676 TensorMemoryType::kStrongShape; in AllocateMemoryForStrongShapes()697 TensorMemoryType::kStrongShape) { in AllocateMemoryForStrongShapes()
104 enum class TensorMemoryType { kStrongShape, kBuffer, kVariable, kConst }; enum139 TensorMemoryType GetTensorMemoryType(const GpuInfo& gpu_info, ValueId id);