| 12 | |
| 13 | namespace { |
| 14 | TestArg generate_transpose_args( |
| 15 | size_t batch, size_t m, size_t n, size_t c, DType dtype) { |
| 16 | TestArg arg; |
| 17 | arg.src = TensorLayout( |
| 18 | TensorShape{batch, n, m, c}, |
| 19 | {static_cast<std::ptrdiff_t>(n * m * c), static_cast<std::ptrdiff_t>(c), |
| 20 | static_cast<std::ptrdiff_t>(n * c), 1}, |
| 21 | dtype); |
| 22 | arg.dst = TensorLayout(TensorShape{batch, n, m, c}, dtype); |
| 23 | return arg; |
| 24 | } |
| 25 | } // anonymous namespace |
| 26 | |
| 27 | namespace megdnn { |
no test coverage detected