Searched refs:expected_engines (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | tf_trt_integration_test_base.py | 588 def _VerifyConnections(self, expected_engines, original_gdef, converted_gdef): argument
593 for engine_name, node_names in expected_engines.items():
678 expected_engines, argument
697 self.assertEqual(len(expected_max_batch_sizes), len(expected_engines))
747 self.assertIn(engine_name, expected_engines)
792 self.assertCountEqual(expected_engines, tuple(name_to_engines_map.keys()))
818 expected_engines = self.ExpectedEnginesToBuild(run_params)
833 self._RemoveGraphSequenceNumber(node.name), expected_engines)
852 self.assertEqual(num_engines, len(expected_engines))
853 if isinstance(expected_engines, dict):
[all …]
|