Searched defs:_partition_linear_fn (Results 1 – 1 of 1) sorted by relevance
102 def _partition_linear_fn(self, name, module, device_mesh): member in ColwiseParallel200 def _partition_linear_fn(self, name, module, device_mesh): member in RowwiseParallel