Function test_ones

tests/python/relax/test_tvmscript_parser_op_create.py:70–82

()

def test_ones():
    @R.function
    def foo(dumb_param: R.Tensor()) -> R.Tensor((2, 3), "float32"):
        gv: R.Tensor((2, 3), "float32") = R.ones((2, 3), "float32")
        return gv

    bb = relax.BlockBuilder()
    dumb_param = relax.Var("dumb_param", R.Tensor())
    with bb.function("foo", [dumb_param]):
        gv = bb.emit(relax.op.ones((2, 3), "float32"))
        bb.emit_func_output(gv)

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


def test_ones_like():