Searched refs:CustomShardingIterDataPipe (Results 1 – 1 of 1) sorted by relevance
3320 class CustomShardingIterDataPipe(IterDataPipe): class3466 sharded_dp = CustomShardingIterDataPipe(dp)3507 dp0 = CustomShardingIterDataPipe(dp0)