()
| 33 | |
| 34 | |
| 35 | def test_create_shape_transposed(): |
| 36 | s = migraphx.shape(lens=[1, 64, 3, 3], permutation=[0, 2, 3, 1]) |
| 37 | assert not s.standard() |
| 38 | assert s.packed() |
| 39 | assert s.transposed() |
| 40 | assert s.lens() == [1, 64, 3, 3] |
| 41 | assert s.strides() == [64 * 3 * 3, 1, 64 * 3, 64] |
| 42 | assert s.ndim() == 4 |
| 43 | |
| 44 | |
| 45 | def test_create_shape_broadcast(): |