Home
last modified time | relevance | path

Searched defs:param_to_hook_all_reduce (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nn/parallel/
H A Ddistributed.py642 param_to_hook_all_reduce=None, argument
969 param_to_hook_all_reduce, argument