Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A D_backward_state.py4 class BackwardState: class
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Ddistributed.py319 def _in_graph_bw_hooks(bw_state: BackwardState):