テンソル バッファーの位置を取得します。
プロトタイプ
location_t get_location();
パラメーター
なし。
戻り値
テンソル バッファーの位置、location_t enum 型の値。
次の表に、location_t enum 型を示します。
名称 | 値 | 説明 |
---|---|---|
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;
}