Home
last modified time | relevance | path

Searched defs:zero_grad (Results 1 – 10 of 10) sorted by relevance

/external/pytorch/torch/distributed/optim/
Dpost_localSGD_optimizer.py106 def zero_grad(self, set_to_none: bool = True): # type: ignore[override] member in PostLocalSGDOptimizer
/external/pytorch/torch/distributed/_shard/sharded_optim/
Dapi.py44 def zero_grad(self, set_to_none: bool = True): # type: ignore[override] member in ShardedOptimizer
/external/pytorch/torch/csrc/jit/mobile/train/optim/
Dsgd.cpp68 void SGD::zero_grad() { in zero_grad() function in torch::jit::mobile::SGD
/external/pytorch/torch/csrc/api/src/optim/
Doptimizer.cpp123 void Optimizer::zero_grad(bool set_to_none) { in zero_grad() function in torch::optim::Optimizer
/external/pytorch/torch/optim/
Doptimizer.py917 def zero_grad(self, set_to_none: bool = True) -> None: member in Optimizer
/external/pytorch/torch/csrc/api/src/nn/
Dmodule.cpp254 void Module::zero_grad(bool set_to_none) { in zero_grad() function in torch::nn::Module
/external/pytorch/torch/distributed/nn/api/
Dremote_module.py449 def zero_grad(self, set_to_none: bool = True) -> None: member in _RemoteModule
/external/pytorch/torch/nn/modules/
Dmodule.py2890 def zero_grad(self, set_to_none: bool = True) -> None: member in Module
/external/pytorch/torch/testing/_internal/distributed/
Ddistributed_test.py4231 zero_grad=False, argument
/external/pytorch/torch/csrc/autograd/
DFunctionsManual.cpp5391 auto zero_grad = at::zeros_symint(zero_grad_shape, input_.options()); in householder_product_backward() local