Searched refs:calculate_shards (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/tools/test/ |
| H A D | test_test_selections.py | 16 from tools.testing.test_selections import calculate_shards, THRESHOLD
107 calculate_shards(2, self.tests, {}, {}, sort_by_time=False),
130 calculate_shards(
165 calculate_shards(
207 calculate_shards(2, self.tests, self.test_times, self.test_class_times),
239 calculate_shards(1, tests, self.test_times, self.test_class_times),
271 calculate_shards(5, self.tests, self.test_times, self.test_class_times),
305 calculate_shards(
359 calculate_shards(
375 calculate_shards(
[all …]
|
| /aosp_15_r20/external/pytorch/tools/testing/ |
| H A D | test_selections.py | 185 def calculate_shards( function
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | run_test.py | 67 calculate_shards,
1578 shards = calculate_shards(
|