下载
EN
注册

TensorPlacement

enum TensorPlacement {
    kOnDeviceHbm,  ///< Tensor位于Device
    kOnHost,       ///< Tensor位于Host
    kFollowing,    ///< Tensor位于Host,且数据紧跟在结构体后面
    kOnDeviceP2p,  ///< Tensor位于Device, P2p内存指的是Device内存透到PCIE BAR空间上,可以让NPU跨PCIE访问
    kTensorPlacementEnd
};