| 6 | namespace imperative { |
| 7 | |
| 8 | BlobManagerImpl::BlobData::BlobData(OwnedBlob* in_blob) { |
| 9 | blob = in_blob; |
| 10 | DeviceTensorStorage d_storage; |
| 11 | d_storage.reset(blob->m_comp_node, blob->m_size, blob->m_storage); |
| 12 | h_storage = HostTensorStorage(blob->m_comp_node); |
| 13 | h_storage.ensure_size(blob->m_size); |
| 14 | h_storage.copy_from(const_cast<DeviceTensorStorage&>(d_storage), blob->m_size); |
| 15 | } |
| 16 | |
| 17 | void BlobManagerImpl::register_blob(OwnedBlob* blob) { |
| 18 | // add blob into the comp2blobs map |