Searched refs:test_all_reduce_sum (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ | ||
| H A D | tpu_strategy_test.py | 1334 def test_all_reduce_sum(self): member in TPUStrategyDistributionTest |
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ | ||
| H A D | distributed_test.py | 2692 def test_all_reduce_sum(self): member in DistributedTest._DistTestBase |