Searched defs:GetMaxBatchSize (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/test/ | ||
| H A D | combined_nms_test.py | 113 def GetMaxBatchSize(self, run_params): member in CombinedNmsExecuteNativeSegmentTest |
| H A D | biasadd_matmul_test.py | 113 def GetMaxBatchSize(self, run_params): member in BiasaddMatMulTest |
| H A D | trt_mode_test.py | 60 def GetMaxBatchSize(self, run_params): member in TrtModeTestBase |
| H A D | annotate_max_batch_sizes_test.py | 58 def GetMaxBatchSize(self, run_params): member in MaxBatchSizesTestBase |
| H A D | int32_test.py | 51 def GetMaxBatchSize(self, run_params): member in ExcludeUnsupportedInt32Test |
| H A D | tf_trt_integration_test_base.py | 291 def GetMaxBatchSize(self, run_params): member in TfTrtIntegrationTestBase |
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ | ||
| H A D | op_builder.h | 360 int GetMaxBatchSize() const { return max_size_for_batch_; } in GetMaxBatchSize() function |