Home
last modified time | relevance | path

Searched refs:IterableWrapper (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_datapipe.py459 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))
[all …]
H A Dtest_dataloader.py57 from torch.utils.data.datapipes.iter import IterableWrapper
1898 datapipe: IterDataPipe = IterableWrapper([[1, 2, 3, 4], [1, 2, 3, 4, 5, 6]])
2949 input_ds = dp.iter.IterableWrapper(exp)
/aosp_15_r20/external/pytorch/test/profiler/
H A Dtest_record_function.py78 input_dp1 = dp.iter.IterableWrapper(range(4))
79 input_dp2 = dp.iter.IterableWrapper(range(4, 8))
80 input_dp3 = dp.iter.IterableWrapper(range(8, 12))
138 input_dp = dp.iter.IterableWrapper(range(10))
/aosp_15_r20/external/jackson-databind/src/test/java/com/fasterxml/jackson/databind/ser/
H A DTestIterable.java14 final static class IterableWrapper class in TestIterable
19 public IterableWrapper(int[] values) { in IterableWrapper() method in TestIterable.IterableWrapper
135 MAPPER.writeValueAsString(new IterableWrapper(new int[] { 1, 2, 3 }))); in testIterable()
/aosp_15_r20/external/pytorch/torch/utils/data/datapipes/iter/
H A D__init__.py38 IterableWrapperIterDataPipe as IterableWrapper, unknown
/aosp_15_r20/external/pytorch/torch/utils/data/datapipes/
H A DREADME.md26 … from the instance, the data may have already changed. Please take [`IterableWrapper`](https://git…