Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dcustom_function.cpp540 bool AutogradContext::needs_input_grad(size_t output_edge_index) const { in needs_input_grad() function in torch::autograd::AutogradContext
546 bool AutogradContext::needs_input_grad( in needs_input_grad() function in torch::autograd::AutogradContext
H A Dpython_function.h98 PyObject* needs_input_grad; member
H A Dpython_function.cpp865 THPObjectPtr needs_input_grad; member
/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dmisc.py831 needs_input_grad=None, argument