Searched defs:zero_grad (Results 1 – 10 of 10) sorted by relevance
106 def zero_grad(self, set_to_none: bool = True): # type: ignore[override] member in PostLocalSGDOptimizer
44 def zero_grad(self, set_to_none: bool = True): # type: ignore[override] member in ShardedOptimizer
68 void SGD::zero_grad() { in zero_grad() function in torch::jit::mobile::SGD
123 void Optimizer::zero_grad(bool set_to_none) { in zero_grad() function in torch::optim::Optimizer
917 def zero_grad(self, set_to_none: bool = True) -> None: member in Optimizer
254 void Module::zero_grad(bool set_to_none) { in zero_grad() function in torch::nn::Module
449 def zero_grad(self, set_to_none: bool = True) -> None: member in _RemoteModule
2890 def zero_grad(self, set_to_none: bool = True) -> None: member in Module
4231 zero_grad=False, argument
5391 auto zero_grad = at::zeros_symint(zero_grad_shape, input_.options()); in householder_product_backward() local