Home
last modified time | relevance | path

Searched refs:backward_c_function (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Dexternal_utils.py97 def call_backward(backward_c_function, saved_tensors, *args): argument
98 fake = FakeBackwardCFunction(backward_c_function, saved_tensors)
H A Dcompiled_autograd.py158 backward_c_function = self.hooks_proxy[backward_idx] # type: ignore[index]
163 backward_c_function,