Searched defs:_shard_tensor (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/distributed/_shard/ | ||
| D | api.py | 16 def _shard_tensor( function |
| /external/pytorch/torch/distributed/tensor/ | ||
| D | placement_types.py | 156 def _shard_tensor( member in Shard |