Function test_resize2d

tests/python/relax/test_tvmscript_parser_op_image.py:37–49

()

def test_resize2d():
    @R.function
    def foo(x: R.Tensor((2, 14, 14, 3), "float32")) -> R.Tensor((2, 28, 28, 3), "float32"):
        gv: R.Tensor((2, 28, 28, 3), "float32") = R.image.resize2d(x, size=(28, 28), layout="NHWC")
        return gv

    bb = relax.BlockBuilder()
    x = relax.Var("x", R.Tensor((2, 14, 14, 3), "float32"))
    with bb.function("foo", [x]):
        gv = bb.emit(relax.op.image.resize2d(x, (28, 28), layout="NHWC"))
        bb.emit_func_output(gv)

    _check(foo, bb.get()["foo"])


def test_resize3d():