Searched refs:set_reduce_ex_hook (Results 1 – 3 of 3) sorted by relevance
68 cls.set_reduce_ex_hook(reduce_hook)80 cls.set_reduce_ex_hook(None)
56 def set_reduce_ex_hook(cls, hook_fn: Any) -> None: ...83 def set_reduce_ex_hook(cls, hook_fn: Any) -> None: ...
206 def set_reduce_ex_hook(cls, hook_fn): member in IterDataPipe345 def set_reduce_ex_hook(cls, hook_fn): member in MapDataPipe