GetPlacement

函数功能

获取tensor的placement，tensor数据所在的设备位置。

// tensor数据所在的设备位置
enum TensorPlacement {
  kOnDeviceHbm,  ///< Tensor位于Device    
  kOnHost,       ///< Tensor位于Host   
  kFollowing,    ///< Tensor位于Host，且数据紧跟在结构体后面    
  kTensorPlacementEnd
};

函数原型

TensorPlacement GetPlacement() const

参数说明

无。

返回值说明

tensor的placement

关于TensorPlacement类型的定义，请参见TensorPlacement。

约束说明

无。

调用示例

auto addr = reinterpret_cast<void *>(0x10);
TensorData td(addr, HostAddrManager, 100U, kOnHost);
auto td_place = td.GetPlacement(); // kOnHost