デスティネーション バッファーにデータをコピーします。
プロトタイプ
void copy_to_host(size_t batch_idx, void* buf, size_t size, size_t offset);
パラメーター
次の表に、copy_to_host
関数の引数を示します。
タイプ | 名称 | 説明 |
---|---|---|
size_t | batch_idx | バッチ インデックス |
void* | buf | デスティネーション バッファー開始アドレス |
size_t | size | コピーするデータのサイズ |
size_t | offset | コピーする開始オフセット |
戻り値
なし。
使用法
vart::TensorBuffer* tb_from;
vart::TensorBuffer* tb_to;
for (auto batch = 0u; batch < batch_size; ++batch) {
std::tie(data, tensor_size) = tb_to->data({(int)batch, 0, 0, 0});
tb_from->copy_to_host(batch, reinterpret_cast<void*>(data),
tensor_size, 0u);
}