コード例 #1
0
ファイル: test_py_converter.py プロジェクト: jiajuns/tvm 
def test_create_empty_tuple():
    empty = relay.Tuple([])
    tup_val = run_as_python(empty)
    assert_adt_len(tup_val, 0)
コード例 #2
0
ファイル: test_py_converter.py プロジェクト: jiajuns/tvm 
def test_create_scalar():
    scalar = relay.const(1)
    tensor_val = run_as_python(scalar)
    print(type(tensor_val))
    assert_tensor_value(tensor_val, 1)
コード例 #3
0
ファイル: test_py_converter.py プロジェクト: jiajuns/tvm 
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)
コード例 #4
0
ファイル: test_py_converter.py プロジェクト: jiajuns/tvm 
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)
コード例 #5
0
ファイル: test_py_converter.py プロジェクト: jiajuns/tvm 
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)
コード例 #6
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]])
コード例 #7
0
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)