| 55 | } |
| 56 | |
| 57 | size_t megdnn::test::WorkspaceBundle::total_size_in_bytes() const { |
| 58 | //! return 0 if the WorkspaceBundle is empty |
| 59 | size_t size = std::accumulate( |
| 60 | m_aligned_sizes.begin(), m_aligned_sizes.end(), static_cast<size_t>(0)); |
| 61 | return size ? size + m_align_in_bytes : size; |
| 62 | } |
| 63 | |
| 64 | std::shared_ptr<TensorNDArray> megdnn::test::dnn_alloc_tensors( |
| 65 | megdnn::Handle* handle, const TensorLayoutArray& layouts, const size_t offset) { |