Home
last modified time | relevance | path

Searched refs:functional_datapipe (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/pytorch/torch/utils/data/datapipes/dataframe/
H A Ddatapipes.py4 from torch.utils.data.datapipes._decorator import functional_datapipe
19 @functional_datapipe("_dataframes_as_tuples")
30 @functional_datapipe("_dataframes_per_row", enable_df_api_tracing=True)
42 @functional_datapipe("_dataframes_concat", enable_df_api_tracing=True)
59 @functional_datapipe("_dataframes_shuffle", enable_df_api_tracing=True)
83 @functional_datapipe("_dataframes_filter", enable_df_api_tracing=True)
111 @functional_datapipe("_to_dataframes_pipe", enable_df_api_tracing=True)
H A Ddataframes.py4 from torch.utils.data.datapipes._decorator import functional_datapipe
441 @functional_datapipe("trace_as_dataframe")
/aosp_15_r20/external/pytorch/torch/utils/data/datapipes/map/
H A Dcombining.py4 from torch.utils.data.datapipes._decorator import functional_datapipe
13 @functional_datapipe("concat")
61 @functional_datapipe("zip")
H A Dcallable.py4 from torch.utils.data.datapipes._decorator import functional_datapipe
22 @functional_datapipe("map")
H A Dgrouping.py4 from torch.utils.data.datapipes._decorator import functional_datapipe
14 @functional_datapipe("batch")
/aosp_15_r20/external/pytorch/torch/utils/data/datapipes/iter/
H A Dcombining.py19 from torch.utils.data.datapipes._decorator import functional_datapipe
37 @functional_datapipe("concat")
77 @functional_datapipe("fork")
386 @functional_datapipe("demux")
601 @functional_datapipe("mux")
670 @functional_datapipe("zip")
H A Dgrouping.py17 from torch.utils.data.datapipes._decorator import functional_datapipe
46 @functional_datapipe("batch")
109 @functional_datapipe("unbatch")
162 @functional_datapipe("groupby")
H A Dcallable.py7 from torch.utils.data.datapipes._decorator import functional_datapipe
25 @functional_datapipe("map")
180 @functional_datapipe("collate")
H A Dstreamreader.py4 from torch.utils.data.datapipes._decorator import functional_datapipe
11 @functional_datapipe("read_from_stream")
H A Dfileopener.py5 from torch.utils.data.datapipes._decorator import functional_datapipe
15 @functional_datapipe("open_files")
H A Dfilelister.py4 from torch.utils.data.datapipes._decorator import functional_datapipe
13 @functional_datapipe("list_files")
H A Drouteddecoder.py4 from torch.utils.data.datapipes._decorator import functional_datapipe
18 @functional_datapipe("routed_decode")
H A Dselecting.py4 from torch.utils.data.datapipes._decorator import functional_datapipe
21 @functional_datapipe("filter")
H A Dsharding.py5 from torch.utils.data.datapipes._decorator import functional_datapipe
31 @functional_datapipe("sharding_filter")
H A Dcombinatorics.py6 from torch.utils.data.datapipes._decorator import functional_datapipe
60 @functional_datapipe("shuffle")
/aosp_15_r20/external/pytorch/torch/utils/data/datapipes/
H A DREADME.md58 Each DataPipe can be registered to support functional API using the decorator `functional_datapipe`.
60 @functional_datapipe("map")
84 @functional_datapipe("parse_csv_files")
H A D_decorator.py13 class functional_datapipe: class
/aosp_15_r20/external/pytorch/torch/utils/data/
H A D__init__.py10 functional_datapipe,