def test_first_input_coeff_not_1(self): graph = build_graph(nodes_attributes, [ ('placeholder_1', 'eltwise'), ('placeholder_2', 'eltwise'), ('eltwise', 'sigmoid_1'), ('eltwise', 'sigmoid_2'), ], {'eltwise': { 'coeff': np.array([2.0, 1.0]) }}, nodes_with_edges_only=True) graph.stage = 'front' graph_ref = build_graph(nodes_attributes, [ ('placeholder_1', 'mul_1'), ('const_1', 'mul_1'), ('mul_1', 'eltwise'), ('placeholder_2', 'eltwise'), ('eltwise', 'sigmoid_1'), ('eltwise', 'sigmoid_2'), ], { 'const_1': { 'value': const_1_value, 'shape': const_1_value.shape }, }, nodes_with_edges_only=True) EltwiseAddNormalize().find_and_replace_pattern(graph) (flag, resp) = compare_graphs(graph, graph_ref, 'placeholder_1', check_op_attrs=True) self.assertTrue(flag, resp)
def test_both_input_coeff_not_defined(self): graph = build_graph(nodes_attributes, [ ('placeholder_1', 'eltwise'), ('placeholder_2', 'eltwise'), ('eltwise', 'sigmoid_1'), ('eltwise', 'sigmoid_2'), ], {'eltwise': { 'coeff': None }}, nodes_with_edges_only=True) graph.stage = 'front' graph_ref = build_graph(nodes_attributes, [ ('placeholder_1', 'eltwise'), ('placeholder_2', 'eltwise'), ('eltwise', 'sigmoid_1'), ('eltwise', 'sigmoid_2'), ], {}, nodes_with_edges_only=True) EltwiseAddNormalize().find_and_replace_pattern(graph) (flag, resp) = compare_graphs(graph, graph_ref, 'placeholder_1', check_op_attrs=True) self.assertTrue(flag, resp)