def test_liveness6(vww_graph): tfi = TfliteImporter() G = tfi.create_graph(vww_graph, {}) assert G steps = add_dimensions(G) liveness = calculate_liveness(G, steps) assert len(liveness) == 121 # no record for 1 output
def add_dimensions(self, quiet=False): if not quiet: LOG.info("update graph dimensions") self.graph_state.steps = add_dimensions(self) if not quiet: LOG.info("calculate liveness") self.graph_state.liveness = calculate_liveness(self, self.graph_state.steps)
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
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
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
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
def test_add_dimension4(cifar10_graph): G = create_graph(cifar10_graph, {}) steps = add_dimensions(G) verify_steps(steps, 16)
def test_add_dimension3(ssd_graph): G = create_graph(ssd_graph, {}) steps = add_dimensions(G) verify_steps(steps, 40)
def test_add_dimension4(cifar10_graph): tfi = TfliteImporter() G = tfi.create_graph(cifar10_graph, {}) steps = add_dimensions(G) verify_steps(steps, 16)
def test_add_dimension1(mnist_graph): G = create_graph(mnist_graph, {}) steps = add_dimensions(G) verify_steps(steps, 10)
def test_liveness2(ir_graph): tfi = TfliteImporter() G = tfi.create_graph(ir_graph, {}) steps = add_dimensions(G) liveness = calculate_liveness(G, steps) assert len(liveness) == 23 # no record for 8 outputs
def test_add_dimension8(mn3_graph): tfi = TfliteImporter() G = tfi.create_graph(mn3_graph, {}) assert len(list(G.dfs())) == 160 steps = add_dimensions(G) verify_steps(steps, 160)
def test_add_dimension7(qvww_graph): tfi = TfliteImporter() G = tfi.create_graph(qvww_graph, {}) steps = add_dimensions(G) verify_steps(steps, 159)
def test_add_dimension5(kws_graph): tfi = TfliteImporter() G = tfi.create_graph(kws_graph, {}) steps = add_dimensions(G) verify_steps(steps, 9)
def add_dimensions(self): LOG.info("update graph dimensions") self.graph_state.steps = add_dimensions(self) LOG.info("calculate liveness") self.graph_state.liveness = calculate_liveness(self, self.graph_state.steps)
def test_add_dimension2(ir_graph): G = create_graph(ir_graph, {}) steps = add_dimensions(G) verify_steps(steps, 31)
def test_load12(): tfi = TfliteImporter() G = tfi.create_graph("tests/graph/imu.tflite", {'load_tensors': True}) steps = add_dimensions(G) verify_steps(steps, 8) assert G