Copy data from the source
buffer.
Prototype
void copy_from_host(size_t batch_idx, const void* buf, size_t size, size_t offset);
Parameters
The following table lists thecopy_from_host
function arguments.
Type | Name | Description |
---|---|---|
size_t | batch_idx | The batch index |
const void* | buf | Source buffer start address |
size_t | size | Data size to be copied |
size_t | offset | The start offset to be copied |
Returns
None.
Usage
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);
}