Searched defs:set_grad (Results 1 – 2 of 2) sorted by relevance
870 def test_to(self, device, dtype, module_info, training, swap, set_grad): argument887 def _to(m, set_grad=False): argument
397 const auto set_grad = [&](Value* x, Value* dx) { in addReverseInline() local