Esempio n. 1
0
def test_fusions1(mnist_graph):
    G = create_graph(mnist_graph, {})
    G.add_dimensions()
    matcher = MatchAllGapConv()
    matcher.match(G)
    G.add_dimensions()
Esempio n. 2
0
def test_adjust4(cifar10_graph):
    G = create_graph(cifar10_graph, {'load_tensors': True})
    G.add_dimensions()
    G.adjust_order()
Esempio n. 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()
Esempio n. 4
0
def test_adjust1(mnist_graph):
    G = create_graph(mnist_graph, {'load_tensors': True})
    G.add_dimensions()
    G.adjust_order()
Esempio n. 5
0
def test_load1(mnist_graph):
    G = create_graph(mnist_graph, {})
    assert G
Esempio n. 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
Esempio n. 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
Esempio n. 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
Esempio n. 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
Esempio n. 10
0
def test_add_dimension3(ssd_graph):
    G = create_graph(ssd_graph, {})
    steps = add_dimensions(G)
    verify_steps(steps, 40)
Esempio n. 11
0
def test_add_dimension4(cifar10_graph):
    G = create_graph(cifar10_graph, {})
    steps = add_dimensions(G)
    verify_steps(steps, 16)
Esempio n. 12
0
def test_add_dimension2(ir_graph):
    G = create_graph(ir_graph, {})
    steps = add_dimensions(G)
    verify_steps(steps, 31)
Esempio n. 13
0
def test_add_dimension1(mnist_graph):
    G = create_graph(mnist_graph, {})
    steps = add_dimensions(G)
    verify_steps(steps, 10)
Esempio n. 14
0
def test_load4(cifar10_graph):
    G = create_graph(cifar10_graph, {})
    assert G
Esempio n. 15
0
def test_load3(ssd_graph):
    G = create_graph(ssd_graph, {})
    assert G
Esempio n. 16
0
def test_load2(ir_graph):
    G = create_graph(ir_graph, {})
    assert G