Home
last modified time | relevance | path

Searched defs:ShardedTest (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/tools/test/
H A Dtest_test_selections.py71 expected_shards: list[tuple[float, list[ShardedTest]]],
486 def comparator(a: ShardedTest, b: ShardedTest) -> int:
504 def comparator(a: ShardedTest, b: ShardedTest) -> int:
/aosp_15_r20/external/pytorch/test/
H A Drun_test.py382 test_module: ShardedTest,
1005 test: ShardedTest, file_path: str, failed: bool, was_rerun: bool
1075 def run_ci_sanity_check(test: ShardedTest, test_directory, options):
1368 def must_serial(file: Union[str, ShardedTest]) -> bool: argument
1595 test: ShardedTest, test_directory: str, options
1624 selected_tests: List[ShardedTest],
/aosp_15_r20/external/pytorch/tools/testing/
H A Dtest_selections.py133 pytest_sharded_tests: Sequence[ShardedTest],
147 sharded_jobs: list[ShardJob], test: ShardedTest
157 tests: Sequence[ShardedTest], sharded_jobs: list[ShardJob]
171 tests: Sequence[ShardedTest], sharded_jobs: list[ShardJob]
H A Dtest_run.py235 class ShardedTest: class