Searched defs:params_with_grad (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/utils/ |
| H A D | clip_grad.h | 26 std::vector<Tensor> params_with_grad; variable
|
| /aosp_15_r20/external/pytorch/torch/optim/ |
| H A D | adamax.py | 87 self, group, params_with_grad, grads, exp_avgs, exp_infs, state_steps argument
|
| H A D | asgd.py | 86 def _init_group(self, group, params_with_grad, grads, mus, axs, etas, state_steps): argument
|
| H A D | radam.py | 91 self, group, params_with_grad, grads, exp_avgs, exp_avg_sqs, state_steps argument
|
| H A D | rmsprop.py | 94 params_with_grad, argument
|
| H A D | _adafactor.py | 72 params_with_grad, argument
|
| H A D | nadam.py | 107 params_with_grad, argument
|
| H A D | adagrad.py | 123 def _init_group(self, group, params_with_grad, grads, state_sums, state_steps): argument
|
| H A D | adamw.py | 112 params_with_grad, argument
|
| H A D | adam.py | 117 params_with_grad, argument
|