Searched defs:datapipe (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/utils/data/datapipes/ |
| H A D | _hook_iterator.py | 46 def _generate_iterdatapipe_msg(datapipe, simplify_dp_name: bool = False): argument
55 def _gen_invalid_iterdatapipe_msg(datapipe): argument
70 def _check_iterator_valid(datapipe, iterator_id, next_method_exists=False) -> None: argument
104 def _set_datapipe_valid_iterator_id(datapipe): argument
129 def profiler_record_fn_context(datapipe): argument
144 def __init__(self, iterator, datapipe, iterator_id, has_next_method): argument
|
| H A D | datapipe.py | 368 def __init__(self, datapipe): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_datapipe.py | 556 def _helper(datapipe): argument
753 def _serialization_test_helper(self, datapipe, use_dill): argument
2107 def _serialization_test_helper(self, datapipe, use_dill): argument
3805 def _yield_count_test_helper(self, datapipe, n_expected_samples): argument
3959 self, datapipe, fast_forward_fn, expected_res, n_iterations=3, rng=None argument
4056 def _snapshot_test_helper(self, datapipe, expected_res, n_iter=3, rng=None): argument
|
| /aosp_15_r20/external/pytorch/torch/utils/data/datapipes/iter/ |
| H A D | streamreader.py | 32 def __init__(self, datapipe, chunk=None): argument
|
| /aosp_15_r20/external/libchrome/mojo/public/java/system/javatests/src/org/chromium/mojo/bindings/ |
| H A D | ReadAndDispatchMessageTest.java | 58 Pair<DataPipe.ProducerHandle, DataPipe.ConsumerHandle> datapipe = core.createDataPipe(null); in setUp() local
|
| /aosp_15_r20/external/pytorch/test/profiler/ |
| H A D | test_record_function.py | 123 def __init__(self, datapipe): argument
|