Searched refs:reduce_ex_hook (Results 1 – 2 of 2) sorted by relevance
123 reduce_ex_hook: Optional[Callable] = None variable in IterDataPipe192 if IterDataPipe.reduce_ex_hook is not None:194 return IterDataPipe.reduce_ex_hook(self)207 if IterDataPipe.reduce_ex_hook is not None and hook_fn is not None:209 IterDataPipe.reduce_ex_hook = hook_fn278 reduce_ex_hook: Optional[Callable] = None variable in MapDataPipe331 if MapDataPipe.reduce_ex_hook is not None:333 return MapDataPipe.reduce_ex_hook(self)346 if MapDataPipe.reduce_ex_hook is not None and hook_fn is not None:348 MapDataPipe.reduce_ex_hook = hook_fn
38 reduce_ex_hook: Optional[Callable] = ...61 reduce_ex_hook: Optional[Callable] = ...