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