Searched defs:any_inputs_require_grad (Results 1 – 2 of 2) sorted by relevance
68 const bool any_inputs_require_grad = params.query.requires_grad() || in input_requires_grad() local
613 …const bool any_inputs_require_grad = query.requires_grad() || key.requires_grad() || value.require… in should_compute_logsumexp() local