| 219 | } |
| 220 | |
| 221 | size_t Image2DTensorFormatBase::image_height(const TensorLayout& layout) const { |
| 222 | size_t accum = 1; |
| 223 | for (int i = m_align_axis - 1; i >= 0; --i) { |
| 224 | if (layout.stride[i] == 0) { |
| 225 | // this dimension is broadcasted |
| 226 | } else { |
| 227 | accum *= layout.shape[i]; |
| 228 | } |
| 229 | } |
| 230 | return accum; |
| 231 | } |
| 232 | |
| 233 | size_t Image2DTensorFormatBase::image_width_elems(const TensorLayout& layout) const { |
| 234 | size_t high_elem = 0; |
no outgoing calls