Method main

tests/python/relax/test_frontend_from_fx.py:621–630

(inp_0: R.Tensor((1, 8, 10, 15), dtype="float32"))

    class expected:
        @R.function
        def main(inp_0: R.Tensor((1, 8, 10, 15), dtype="float32")) -> R.Tensor(
            (1, 2, 20, 30), dtype="float32"
        ):
            with R.dataflow():
                lv: R.Tensor((1, 2, 20, 30), dtype="float32") = R.nn.pixel_shuffle(
                    inp_0, upscale_factor=2
                )
                gv: R.Tensor((1, 2, 20, 30), dtype="float32") = lv
                R.output(gv)
            return gv

    input_infos = [([1, 8, 10, 15], "float32")]
    verify_model(PixelShuffle1(2), input_infos, {}, expected)