Searched refs:expected_ops (Results 1 – 2 of 2) sorted by relevance
138 def assert_ops_in_graph(expected_ops, graph): argument154 if node.name in expected_ops:155 if expected_ops[node.name] != node.op:157 (node.name, expected_ops[node.name], node.op))159 if set(expected_ops.keys()) != set(actual_ops.keys()):161 (expected_ops.keys(), actual_ops.keys()))
681 expected_ops = ops_recording_on + ops_recording_off682 self.assertCountEqual(expected_ops, summary_ops.all_v2_summary_ops())