def test_component_map_loading_scale(self): test_map = "input-node name=input dim=16\n" + \ "component-node name=lda component=lda input=Scale(0.1, input)\n" + \ "\n" graph = Graph(name="test_graph_component_map_loading_scale") test_top_map = load_topology_map(io.BytesIO(bytes(test_map, 'ascii')), graph) ref_map = {b"lda": ["lda"]} self.assertEqual(test_top_map, ref_map) self.assertTrue("input" in graph.nodes()) self.assertListEqual(list(Node(graph, 'input')['shape']), [1, 16]) ref_graph = build_graph({'input': {'shape': np.array([1, 16]), 'kind': 'op', 'op': 'Parameter'}, 'lda': {'kind': 'op'}, 'mul': {'kind': 'op'}, 'scale_const': {'kind': 'op', 'op': 'Const'}, }, [ ('input', 'mul', {'in': 0}), ('scale_const', 'mul', {'in': 1}), ('mul', 'lda', {'out': 0}), ] ) (flag, resp) = compare_graphs(graph, ref_graph, 'lda') self.assertTrue(flag, resp)
def test_component_map_loading_swap(self): test_map = "input-node name=input dim=16 \n" + \ "component-node name=lda component=lda input=input \n" + \ "component-node name=tdnn1.batchnorm component=tdnn1.batchnorm input=tdnn1.relu \n" + \ "component-node name=tdnn1.relu component=tdnn1.relu input=tdnn1.affine \n" + \ "component-node name=tdnn1.affine component=tdnn1.affine input=lda \n" + \ "\n" graph = Graph(name="test_graph_component_map_loading_swap") test_top_map = load_topology_map(io.BytesIO(bytes(test_map, 'ascii')), graph) ref_map = {b"lda": ["lda"], b"tdnn1.affine": ["tdnn1.affine"], b"tdnn1.relu": ["tdnn1.relu"], b"tdnn1.batchnorm": ["tdnn1.batchnorm"]} self.assertEqual(test_top_map, ref_map) self.assertTrue("input" in graph.nodes()) self.assertListEqual(list(Node(graph, 'input')['shape']), [1, 16]) ref_graph = build_graph({'input': {'shape': np.array([1, 16]), 'kind': 'op', 'op': 'Parameter'}, 'lda': {'kind': 'op'}, 'tdnn1.affine': {'kind': 'op'}, 'tdnn1.relu': {'kind': 'op'}, 'tdnn1.batchnorm': {'kind': 'op'}, }, [ ('input', 'lda'), ('lda', 'tdnn1.affine'), ('tdnn1.affine', 'tdnn1.relu'), ('tdnn1.relu', 'tdnn1.batchnorm'), ] ) (flag, resp) = compare_graphs(graph, ref_graph, 'tdnn1.batchnorm') self.assertTrue(flag, resp)
def test_component_map_loading_offset(self): test_map = "input-node name=input dim=16\n" + \ "component-node name=lda component=lda input=Offset(input, -3)\n" + \ "component-node name=tdnn1.affine component=tdnn1.affine input=Append(Offset(input, -1), Offset(lda, 1))\n" + \ "component-node name=tdnn1.relu component=tdnn1.relu input=tdnn1.affine\n" + \ "\n" graph = Graph(name="test_graph_component_map_loading_offset") test_top_map= load_topology_map(io.BytesIO(bytes(test_map, 'ascii')), graph) ref_map = {b"lda": ["lda"], b"tdnn1.affine": ["tdnn1.affine"], b"tdnn1.relu": ["tdnn1.relu"]} self.assertEqual(test_top_map, ref_map) self.assertTrue("input" in graph.nodes()) self.assertListEqual(list(Node(graph, 'input')['shape']), [1, 16]) ref_graph = build_graph({'input': {'shape': np.array([1, 16]), 'kind': 'op', 'op': 'Parameter'}, 'lda': {'kind': 'op'}, 'tdnn1.affine': {'kind': 'op'}, 'tdnn1.relu': {'kind': 'op'}, 'append_input_lda': {'kind': 'op', 'op': 'Concat'}, 'offset_in_input_3': {'kind': 'op', 'op': 'memoryoffset', 't': -3, 'pair_name': 'offset_out_input_3'}, 'offset_in_input_1': {'kind': 'op', 'op': 'memoryoffset', 't': -1, 'pair_name': 'offset_out_input_1'}, 'offset_in_lda_1': {'kind': 'op', 'op': 'memoryoffset', 't': -1, 'pair_name': 'offset_out_lda_1'}, }, [ ('input', 'offset_in_input_3', {'out': 0}), ('offset_in_input_3', 'lda', {'out': 0}), ('lda', 'offset_in_lda_1', {'out': 0}), ('input', 'offset_in_input_1', {'out': 1}), ('offset_in_lda_1', 'append_input_lda', {'in': 1, 'out': 0}), ('offset_in_input_1', 'append_input_lda', {'in': 0, 'out': 0}), ('append_input_lda', 'tdnn1.affine', {'out': 0}), ('tdnn1.affine', 'tdnn1.relu', {'out': 0}), ] ) (flag, resp) = compare_graphs(graph, ref_graph, 'tdnn1.relu') self.assertTrue(flag, resp)