| 46 | } |
| 47 | |
| 48 | bool check_tensor_for_transpose( |
| 49 | const TensorLayout& src_layout, const TensorLayout& dst_layout, |
| 50 | std::vector<size_t>& permute) { |
| 51 | if (!src_layout.eq_shape(dst_layout)) { |
| 52 | return false; |
| 53 | } |
| 54 | if (!dst_layout.is_contiguous()) { |
| 55 | return false; |
| 56 | } |
| 57 | size_t ndim = src_layout.ndim; |
| 58 | |
| 59 | permute.resize(ndim); |
| 60 | rep(i, ndim) { permute[i] = i; } |
| 61 | std::sort(permute.begin(), permute.end(), [&src_layout](int i, int j) { |
| 62 | return src_layout.stride[i] >= src_layout.stride[j]; |
| 63 | }); |
| 64 | TensorLayout ori_src_layout = src_layout.dimshuffle(permute); |
| 65 | if (!ori_src_layout.is_contiguous()) { |
| 66 | return false; |
| 67 | } |
| 68 | return true; |
| 69 | } |
| 70 | |
| 71 | bool try_transpose( |
| 72 | _megdnn_tensor_in src, _megdnn_tensor_out dst, RelayoutForwardImpl* opr, |
no test coverage detected