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