(inputs, f, c)
| 1925 | def _get_layerPixelShuffle(device, dtype, dim_order): |
| 1926 | @subgraph("LayerPixelShuffle", dtype, device, 3) |
| 1927 | def layerPixelShuffle(inputs, f, c): |
| 1928 | inp, shape_0, shape_1 = inputs |
| 1929 | inp = f(Reshape(), inp, shape_0) |
| 1930 | inp = f(Dimshuffle(dim_order), inp) |
| 1931 | oup = f(Reshape(), inp, shape_1) |
| 1932 | return (oup,), (True,) |
| 1933 | |
| 1934 | return layerPixelShuffle |
| 1935 |
no test coverage detected