Function test_tensor_struct_info

tests/python/relax/test_tvmscript_printer_relax.py:208–219

()

def test_tensor_struct_info():
    obj = relax.TensorStructInfo(
        shape=relax.ShapeExpr([1, tirx.Var("a", "int64"), 3]),
        dtype="float32",
    )
    _assert_print(
        obj,
        """
a = T.int64()
R.Tensor((1, a, 3), dtype="float32")
""",
    )


def test_tuple_struct_info_empty():