Beispiel #1
0
def test_fusions1(mnist_graph):
    G = create_graph(mnist_graph, {})
    G.add_dimensions()
    matcher = MatchAllGapConv()
    matcher.match(G)
    G.add_dimensions()
Beispiel #2
0
def test_adjust4(cifar10_graph):
    G = create_graph(cifar10_graph, {'load_tensors': True})
    G.add_dimensions()
    G.adjust_order()
Beispiel #3
0
def test_fusions4(ssd_graph):
    G = create_graph(ssd_graph, {})
    G.add_dimensions()
    matcher = get_std_match_group()
    matcher.match(G)
    G.add_dimensions()
Beispiel #4
0
def test_adjust1(mnist_graph):
    G = create_graph(mnist_graph, {'load_tensors': True})
    G.add_dimensions()
    G.adjust_order()
Beispiel #5
0
def test_load1(mnist_graph):
    G = create_graph(mnist_graph, {})
    assert G
Beispiel #6
0
def test_liveness3(ssd_graph):
    G = create_graph(ssd_graph, {})
    assert G
    steps = add_dimensions(G)
    liveness = calculate_liveness(G, steps)
    assert len(liveness) == 39 # no record for 1 output
Beispiel #7
0
def test_liveness4(cifar10_graph):
    G = create_graph(cifar10_graph, {})
    assert G
    steps = add_dimensions(G)
    liveness = calculate_liveness(G, steps)
    assert len(liveness) == 15 # no record for 1 output
Beispiel #8
0
def test_liveness1(mnist_graph):
    G = create_graph(mnist_graph, {})
    steps = add_dimensions(G)
    liveness = calculate_liveness(G, steps)
    assert len(liveness) == 9 # no record for 1 output
Beispiel #9
0
def test_liveness2(ir_graph):
    G = create_graph(ir_graph, {})
    steps = add_dimensions(G)
    liveness = calculate_liveness(G, steps)
    assert len(liveness) == 23 # no record for 8 outputs
Beispiel #10
0
def test_add_dimension3(ssd_graph):
    G = create_graph(ssd_graph, {})
    steps = add_dimensions(G)
    verify_steps(steps, 40)
Beispiel #11
0
def test_add_dimension4(cifar10_graph):
    G = create_graph(cifar10_graph, {})
    steps = add_dimensions(G)
    verify_steps(steps, 16)
Beispiel #12
0
def test_add_dimension2(ir_graph):
    G = create_graph(ir_graph, {})
    steps = add_dimensions(G)
    verify_steps(steps, 31)
Beispiel #13
0
def test_add_dimension1(mnist_graph):
    G = create_graph(mnist_graph, {})
    steps = add_dimensions(G)
    verify_steps(steps, 10)
Beispiel #14
0
def test_load4(cifar10_graph):
    G = create_graph(cifar10_graph, {})
    assert G
Beispiel #15
0
def test_load3(ssd_graph):
    G = create_graph(ssd_graph, {})
    assert G
Beispiel #16
0
def test_load2(ir_graph):
    G = create_graph(ir_graph, {})
    assert G