| 186 | } |
| 187 | |
| 188 | size_t WorkspaceBundle::total_size_in_bytes() const { |
| 189 | //! return 0 if the WorkspaceBundle is empty |
| 190 | size_t size = std::accumulate( |
| 191 | m_aligned_sizes.begin(), m_aligned_sizes.end(), static_cast<size_t>(0)); |
| 192 | return size ? size + m_align_in_bytes : size; |
| 193 | } |
| 194 | |
| 195 | size_t megdnn::count_not_ones_in_shape(const TensorShape& shape) { |
| 196 | size_t res = 0u; |