Searched refs:GetMaxBatchSize (Results 1 – 8 of 8) sorted by relevance
113 def GetMaxBatchSize(self, run_params): member in CombinedNmsExecuteNativeSegmentTest123 return super().GetMaxBatchSize(run_params) - 1
55 def GetMaxBatchSize(self, run_params): member in ExcludeUnsupportedInt32Test
71 def GetMaxBatchSize(self, run_params): member in ImplicitBatchTest
62 def GetMaxBatchSize(self, run_params): member in MaxBatchSizesTestBase
115 def GetMaxBatchSize(self, run_params): member in BiasaddMatMulTest
287 def GetMaxBatchSize(self, run_params): member in TfTrtIntegrationTestBase335 max_batch_size = self.GetMaxBatchSize(run_params)465 max_batch_size=self.GetMaxBatchSize(run_params),860 default_max_batch_size=self.GetMaxBatchSize(run_params))
331 max_sizes[0] = graph_builder_->GetMaxBatchSize(); in AddOutput()351 max_sizes[0] = graph_builder_->GetMaxBatchSize(); in AddOutput()
360 int GetMaxBatchSize() const { return max_size_for_batch_; } in GetMaxBatchSize() function