从源缓冲区复制数据。
原型设计
void copy_from_host(size_t batch_idx, const void* buf, size_t size, size_t offset);参数
下表列出了copy_from_host 函数实参。
| 类型 | 名称 | 描述 |
|---|---|---|
| size_t | batch_idx | 批次索引 |
| const 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_from->data({(int)batch, 0, 0, 0});
tb_to->copy_from_host(batch, reinterpret_cast<const void*>(data),
tensor_size, 0u);
}