Searched defs:apply_sharding (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/utils/data/datapipes/iter/ | ||
| D | sharding.py | 22 def apply_sharding( member in _ShardingIterDataPipe |
| /external/pytorch/torch/utils/data/ | ||
| D | graph_settings.py | 48 def apply_sharding( function |
| /external/pytorch/test/ | ||
| D | test_datapipe.py | 3326 def apply_sharding(self, num_of_instances, instance_id): member in CustomShardingIterDataPipe |