Lines Matching refs:IterableWrapper
459 datapipe1 = dp.iter.IterableWrapper(
543 numbers_dp = dp.iter.IterableWrapper(source_numbers)
818 (dp.iter.Concater, None, (dp.iter.IterableWrapper(range(5)),), {}),
827 (dp.iter.IterableWrapper, range(10), (), {}),
830 (dp.iter.Multiplexer, None, (dp.iter.IterableWrapper(range(10)),), {}),
832 (dp.iter.Shuffler, dp.iter.IterableWrapper([0] * 10), (), {}),
835 (dp.iter.Zipper, None, (dp.iter.IterableWrapper(range(10)),), {}),
844 custom_input = dp.iter.IterableWrapper(range(10))
861 input_dp = dp.iter.IterableWrapper(range(10))
946 input_dp = dp.iter.IterableWrapper(range(10))
978 input_dp = dp.iter.IterableWrapper(input_ls)
993 input_dp_shallow = dp.iter.IterableWrapper(input_ls2, deepcopy=False)
999 input_dp = dp.iter.IterableWrapper(input_ls)
1011 input_dp1 = dp.iter.IterableWrapper(range(10))
1012 input_dp2 = dp.iter.IterableWrapper(range(5))
1046 input_dp = dp.iter.IterableWrapper(range(10))
1216 input_dp1 = dp.iter.IterableWrapper(range(4))
1217 input_dp2 = dp.iter.IterableWrapper(range(4, 8))
1218 input_dp3 = dp.iter.IterableWrapper(range(8, 12))
1225 input_dp1 = dp.iter.IterableWrapper([1, 2, 3, 4])
1226 input_dp2 = dp.iter.IterableWrapper([10])
1227 input_dp3 = dp.iter.IterableWrapper([100, 200, 300])
1234 input_dp1 = dp.iter.IterableWrapper([0, 1, 2, 3])
1235 input_dp2 = dp.iter.IterableWrapper([])
1241 input_dp1 = dp.iter.IterableWrapper(range(10))
1248 input_dp = dp.iter.IterableWrapper(range(10))
1411 input_dp = dp.iter.IterableWrapper(range(target_length))
1485 datapipe = dp.iter.IterableWrapper(
1628 datapipe = dp.iter.IterableWrapper(
1752 input_dp = dp.iter.IterableWrapper(arrs)
1794 input_dp = dp.iter.IterableWrapper(arrs)
1835 prebatch_dp = dp.iter.IterableWrapper(range(target_length))
1845 input_dp = dp.iter.IterableWrapper([[0, 1, 2], [3, 4, 5]])
1851 input_dp = dp.iter.IterableWrapper([[[0, 1], [2, 3]], [[4, 5], [6, 7]]])
1874 input_dp = dp.iter.IterableWrapper([[0, 1, 2], [3, 4, 5]])
1887 input_dp = dp.iter.IterableWrapper([[0, 1, 2], [3, 4, 5]])
1897 input_ds = dp.iter.IterableWrapper(range(10))
1915 tuple_input_ds = dp.iter.IterableWrapper([(d - 1, d, d + 1) for d in range(10)])
1970 input_dp = dp.iter.IterableWrapper(range(10))
1990 input_dp = dp.iter.IterableWrapper(
2008 input_dp = dp.iter.IterableWrapper(list(range(10)))
2075 … dp.iter.Zipper(dp.iter.IterableWrapper(range(10)), list(range(10))) # type: ignore[arg-type]
2079 dp.iter.IterableWrapper(range(10)), IDP_NoLen(range(5))
2090 dp.iter.IterableWrapper(range(10)), dp.iter.IterableWrapper(range(5))
2693 dp0 = DP0(dp.iter.IterableWrapper(range(10)))
2924 iter_dp = dp.iter.IterableWrapper(source_map_dp)
2932 source_iter_dp = dp.iter.IterableWrapper(list(range(10)))
2955 source_iter_dp = dp.iter.IterableWrapper(list(range(10)))
2981 idp = dp.iter.IterableWrapper(range(3)).map(lambda x: x + 1).shuffle()
3024 source_dp if source_dp else dp.iter.IterableWrapper([1, 2, 4])
3472 numbers_dp = dp.iter.IterableWrapper(range(13))
3484 numbers_dp = dp.iter.IterableWrapper(range(1))
3515 numbers_dp = dp.iter.IterableWrapper(range(13))
3524 numbers_dp = dp.iter.IterableWrapper(range(13)).sharding_filter()
3534 numbers_dp = dp.iter.IterableWrapper(range(13))
3544 numbers_dp = dp.iter.IterableWrapper(range(13)).sharding_filter()
3551 numbers_dp = dp.iter.IterableWrapper(range(13))
3590 source_dp: IterDataPipe = dp.iter.IterableWrapper(range(10))
3642 dp.iter.IterableWrapper(range(10)).map(_fake_fn).filter(_fake_filter_fn)
3678 dp.iter.IterableWrapper(range(10)).map(_fake_fn).filter(_fake_filter_fn)
3734 source_dp: IterDataPipe = dp.iter.IterableWrapper(range(10))
3756 source_dp = dp.iter.IterableWrapper(range(10))
3774 dp.iter.IterableWrapper(range(10)).map(_fake_fn).filter(_fake_filter_fn)
3826 datapipe: IterDataPipe = dp.iter.IterableWrapper(range(10))
3984 graph1 = dp.iter.IterableWrapper(range(10))
4081 graph1 = dp.iter.IterableWrapper(range(10))
4120 dp.iter.IterableWrapper(range(10))