Searched defs:set_to_none (Results 1 – 2 of 2) sorted by relevance
123 void Optimizer::zero_grad(bool set_to_none) { in zero_grad()
254 void Module::zero_grad(bool set_to_none) { in zero_grad()