获取张量缓冲区所在位置。
原型设计
location_t get_location();参数
无。
返回
张量缓冲区位置，location_t 枚举类型值。
下表列出了 location_t 枚举类型。
| 名称 | 值 | 描述 |
|---|---|---|
| HOST_VIRT | 0 | 仅可供主机访问。 |
| HOST_PHY | 1 | 连续物理存储器，在主机与器件之间共享。 |
| DEVICE_0 | 2 | 仅可供 device_* 访问。 |
| DEVICE_1 | 3 | |
| DEVICE_2 | 4 | |
| DEVICE_3 | 5 | |
| DEVICE_4 | 6 | |
| DEVICE_5 | 7 | |
| DEVICE_6 | 8 | |
| DEVICE_7 | 9 |
用法
vart::TensorBuffer* tb;
switch (tb->get_location()) {
case vart::TensorBuffer::location_t::HOST_VIRT:
// do nothing
break;
case vart::TensorBuffer::location_t::HOST_PHY:
// do nothing
break;
default:
// do nothing
break;
}