Searched refs:reduce_scatter_with_assert (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/ |
| H A D | test_fully_shard_mixed_precision.py | 26 reduce_scatter_with_assert,
84 reduce_scatter_with_assert, self, orig_reduce_scatter, assert_fn
153 reduce_scatter_with_assert, self, orig_reduce_scatter, assert_fn
197 reduce_scatter_with_assert, self, orig_reduce_scatter, assert_fn
268 reduce_scatter_with_assert, self, orig_reduce_scatter, assert_fn
|
| H A D | test_fully_shard_frozen.py | 24 reduce_scatter_with_assert,
102 reduce_scatter_with_assert, self, orig_reduce_scatter, assert_fn
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_fsdp.py | 1039 def reduce_scatter_with_assert( function
|