文档管理中心
我的
您当前正在浏览新版开发者文档中心,目录分类和层级有所调整。点击左侧当前文档分类名称前的“☰”图标,可切换文档分类。 了解新版目录

TensorPlacement

Tensor存储位置的枚举值定义如下。

收起
自动换行
深色代码主题
复制
  1. enum TensorPlacement {
  2. kOnDeviceHbm, // < Tensor位于Device上的HBM内存
  3. kOnHost, // < Tensor位于Host
  4. kFollowing, // < Tensor位于Host,且数据紧跟在结构体后面
  5. kOnDeviceP2p, // < Tensor位于Device上的P2p内存,指的是HBM透到PCIE BAR空间上,可以让NPU跨PCIE能访问的地址空间
  6. kTensorPlacementEnd
  7. };