用于获取Tensor的内存类型。
1 | MemoryData::MemoryType TensorBase::GetTensorType() const; |
数据结构 |
说明 |
---|---|
MemoryType |
申请的内存类型: ● MEMORY_HOST:对应Host侧。 ● MEMORY_DEVICE:对应Device侧。 ● MEMORY_DVPP:对应DVPP侧。 ● MEMORY_HOST_MALLOC:对应malloc申请内存。 ● MEMORY_HOST_NEW:对应new申请内存。 |