Searched defs:_reduce_shard_value (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/ | ||
| H A D | _embedding_ops.py | 132 def _reduce_shard_value( member in _MaskPartial |
| H A D | _math_ops.py | 110 def _reduce_shard_value( member in _NormPartial |
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/ | ||
| H A D | placement_types.py | 603 def _reduce_shard_value( member in Partial |