Home
last modified time | relevance | path

Searched refs:enable_df_api_tracing (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/utils/data/datapipes/dataframe/
H A Ddatapipes.py30 @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)
/aosp_15_r20/external/pytorch/torch/utils/data/datapipes/
H A D_decorator.py16 def __init__(self, name: str, enable_df_api_tracing=False) -> None: argument
25 self.enable_df_api_tracing = enable_df_api_tracing
44 self.name, cls, enable_df_api_tracing=self.enable_df_api_tracing
H A Ddatapipe.py155 cls, function_name, cls_to_register, enable_df_api_tracing=False argument
162 def class_function(cls, enable_df_api_tracing, source_dp, *args, **kwargs): argument
165 if enable_df_api_tracing or isinstance(source_dp, DFIterDataPipe):
172 class_function, cls_to_register, enable_df_api_tracing
H A Ddatapipe.pyi.in76 enable_df_api_tracing: bool = ...,