| 340 | MemAllocPlan& mem_plan() { return m_mem_plan; } |
| 341 | |
| 342 | bool dev_tensor_valid() const { |
| 343 | return m_mem_plan.valid() && m_mem_plan.layout().eq_shape(m_shape) && |
| 344 | m_dev_tensor.storage().comp_node_valid() && |
| 345 | m_dev_tensor.layout().eq_layout(m_mem_plan.layout()) && |
| 346 | m_dev_tensor.comp_node() == m_comp_node; |
| 347 | } |
| 348 | |
| 349 | //! get the underlying device tensor to fill data |
| 350 | MGE_WIN_DECLSPEC_FUC const DeviceTensorND& dev_tensor() const; |