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