コード例 #1
0
ファイル: axis_test.py プロジェクト: zhangaz1/webdnn 
def test_unify_chain():
    a1 = Axis()
    a2 = Axis()
    a3 = Axis()
    a1.unify(a2)
    a1.unify(a3)
    assert a2 == a3
コード例 #2
0
ファイル: axis_test.py プロジェクト: zhangaz1/webdnn 
def test_unify_deep_chain_different_name_axes():
    a1 = Axis(name="A")
    a2 = Axis()
    a3 = Axis(name="B")
    a4 = Axis()
    a1.unify(a2)
    a3.unify(a4)
    a2.unify(a4)
コード例 #3
0
ファイル: axis_test.py プロジェクト: zhangaz1/webdnn 
def test_unify_deep_chain():
    a1 = Axis()
    a2 = Axis()
    a3 = Axis()
    a4 = Axis()
    a1.unify(a2)
    a3.unify(a4)
    a1.unify(a3)
    assert a2 == a4
コード例 #4
0
ファイル: axis_test.py プロジェクト: zhangaz1/webdnn 
def test_unify_different_name_axes():
    a1 = Axis(name="A")
    a2 = Axis(name="B")
    a1.unify(a2)
コード例 #5
0
ファイル: axis_test.py プロジェクト: zhangaz1/webdnn 
def test_unify_same_name_axes():
    a1 = Axis(name="A")
    a2 = Axis(name="A")
    a1.unify(a2)
    assert a1 == a2
コード例 #6
0
ファイル: axis_test.py プロジェクト: zhangaz1/webdnn 
def test_unify_resolved_axes():
    a1 = Axis()
    a2 = Axis()
    a1.unify(a2)
    a1.unify(a2)
    assert a1 == a2
コード例 #7
0
ファイル: axis_test.py プロジェクト: zhangaz1/webdnn 
def test_unify_same_axis():
    a1 = Axis()
    a1.unify(a1)
    assert a1 == a1
コード例 #8
0
ファイル: axis_test.py プロジェクト: zhangaz1/webdnn 
def test_unify():
    a1 = Axis()
    a2 = Axis()
    a1.unify(a2)
    assert a1 == a2