Searched defs:_partition_embedding_fn (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/parallel/ | ||
| H A D | style.py | 112 def _partition_embedding_fn(self, name, module, device_mesh): member in ColwiseParallel |