Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.cpp1544 const auto self_requires_grad = grad_input_mask[0]; in sparse_sampled_addmm_backward() local
1548 self_requires_grad ? maybe_multiply(grad, beta.conj()) : Tensor{}, in sparse_sampled_addmm_backward()
5544 const auto self_requires_grad = grad_output_mask[0]; in ormqr_backward() local
5551 if (self_requires_grad || tau_requires_grad) { in ormqr_backward()