Searched refs:expected_max_batch_sizes (Results 1 – 1 of 1) sorted by relevance
682 expected_max_batch_sizes=None, argument696 if isinstance(expected_max_batch_sizes, collections.abc.Collection):697 self.assertEqual(len(expected_max_batch_sizes), len(expected_engines))700 expected_max_batch_sizes,793 if expected_max_batch_sizes is not None:794 self.assertCountEqual(expected_max_batch_sizes, actual_max_batch_sizes)859 expected_max_batch_sizes=self.ExpectedMaxBatchSizes(run_params),