def test_create_empty_tuple(): empty = relay.Tuple([]) tup_val = run_as_python(empty) assert_adt_len(tup_val, 0)
def test_create_scalar(): scalar = relay.const(1) tensor_val = run_as_python(scalar) print(type(tensor_val)) assert_tensor_value(tensor_val, 1)
def test_op_add(): add = relay.add(relay.const(1), relay.const(2)) add_val = run_as_python(add) assert_tensor_value(add_val, 3)
def test_ref_read(): v = relay.Var("v") assign = relay.Let(v, relay.RefCreate(relay.Tuple([])), relay.RefRead(v)) read_val = run_as_python(assign) assert_adt_len(read_val, 0)
def test_create_ref(): relay_ref = relay.RefCreate(relay.Tuple([])) ref_val = run_as_python(relay_ref) assert isinstance(ref_val, RefValue) assert_adt_len(ref_val.value, 0)
def test_create_tensor(): tensor = relay.const([[1, 1], [2, 2]]) tensor_val = run_as_python(tensor) assert_tensor_value(tensor_val, [[1, 1], [2, 2]])
def test_ref_read(): v = relay.Var('v') assign = relay.Let(v, relay.RefCreate(relay.Tuple([])), relay.RefRead(v)) read_val = run_as_python(assign) assert_tuple_value(read_val, 0)