| 231 | } |
| 232 | |
| 233 | inline TensorShape cvt_src_or_dst_ncdhw2ndhwc(const TensorShape& shape) { |
| 234 | megdnn_assert(shape.ndim == 5); |
| 235 | auto N = shape[0], C = shape[1], D = shape[2], H = shape[3], W = shape[4]; |
| 236 | return TensorShape{N, D, H, W, C}; |
| 237 | } |
| 238 | |
| 239 | inline TensorShape cvt_filter_nchw2nhwc(const TensorShape& shape) { |
| 240 | if (shape.ndim == 4) { |