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 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_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_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