| 292 | |
| 293 | template <class Trait> |
| 294 | void TensorStorage<Trait>::reset(CompNode node, size_t size, RawStorage data) { |
| 295 | mgb_assert(m_allow_realloc); |
| 296 | m_comp_node = node; |
| 297 | m_size = size; |
| 298 | m_capacity = size; |
| 299 | m_offset = 0; |
| 300 | m_data = std::move(data); |
| 301 | m_ref_ptr = std::make_shared<void*>(static_cast<void*>(m_data.get())); |
| 302 | } |
| 303 | |
| 304 | template <class Trait> |
| 305 | void TensorStorage<Trait>::only_reset_raw_storage( |