Searched defs:_partition_value (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/distributed/tensor/_ops/ | ||
| D | _embedding_ops.py | 87 def _partition_value( member in _MaskPartial |
| D | _math_ops.py | 87 def _partition_value( member in _NormPartial |
| /external/pytorch/torch/distributed/tensor/ | ||
| D | placement_types.py | 615 def _partition_value( member in Partial |