Home
last modified time | relevance | path

Searched defs:test_gather (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_control_collectives.py52 def test_gather(self) -> None: member in TestCollectives
H A Dtest_multi_threaded_pg.py282 def test_gather(self): member in TestCollectivesWithBaseClass
H A Dtest_c10d_spawn_gloo.py278 def test_gather(self): member in TestDistributedNNFunctionsGloo
/aosp_15_r20/external/python/cpython3/Lib/test/test_peg_generator/
Dtest_first_sets.py148 def test_gather(self) -> None: member in TestFirstSets
Dtest_c_parser.py203 def test_gather(self) -> None: member in TestCParser
Dtest_pegen.py78 def test_gather(self) -> None: member in TestPegen
/aosp_15_r20/external/pytorch/test/distributed/_tensor/
H A Dtest_tensor_ops.py428 def test_gather(self): member in DistTensorOpsTest
/aosp_15_r20/external/pytorch/test/
H A Dtest_scatter_gather_ops.py38 def test_gather(self, device, dtype): member in TestScatterGather
H A Dtest_cuda_multigpu.py1666 def test_gather(self): member in TestCudaComm
H A Dtest_mps.py7400 def test_gather(self): member in TestMPS
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_python_builtins.py141 def test_gather(self): member in TestPythonBuiltinOP
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Darray_test.py35 def test_gather(self): member in ArrayTest
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_operators.py822 def test_gather(self): member in TestOperators
H A Dtest_pytorch_onnx_onnxruntime.py4341 def test_gather(self): member in TestONNXRuntime
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Ddistributed_test.py3387 def test_gather(self): member in DistributedTest._DistTestBase