| 627 | } |
| 628 | |
| 629 | VARP _ChannelShuffle(VARP x, int group) { |
| 630 | x = _Convert(x, NHWC); |
| 631 | x = _Reshape(x, {0, 0, 0, group, -1}, NHWC); |
| 632 | x = _Transpose(x, {0, 1, 2, 4, 3}); |
| 633 | x = _Reshape(x, {0, 0, 0, -1}, NHWC); |
| 634 | x = _Convert(x, NC4HW4); |
| 635 | return x; |
| 636 | } |
| 637 | |
| 638 | VARP _Reverse(VARP x, VARP axis) { |
| 639 | std::unique_ptr<MNN::OpT> op(new MNN::OpT); |