Searched defs:_init_group (Results 1 – 13 of 13) sorted by relevance
| /external/pytorch/test/dynamo/ | ||
| D | test_optimizers.py | 19 def _init_group(self, params, group): member in MyOptimizer |
| /external/pytorch/torch/optim/ | ||
| D | adadelta.py | 83 def _init_group( member in Adadelta |
| D | rprop.py | 78 def _init_group(self, group, params, grads, prevs, step_sizes, state_steps): member in Rprop |
| D | adamax.py | 86 def _init_group( member in Adamax |
| D | asgd.py | 86 def _init_group(self, group, params_with_grad, grads, mus, axs, etas, state_steps): member in ASGD |
| D | radam.py | 90 def _init_group( member in RAdam |
| D | sgd.py | 80 def _init_group(self, group, params, grads, momentum_buffer_list): member in SGD |
| D | rmsprop.py | 91 def _init_group( member in RMSprop |
| D | _adafactor.py | 69 def _init_group( member in Adafactor |
| D | nadam.py | 104 def _init_group( member in NAdam |
| D | adagrad.py | 123 def _init_group(self, group, params_with_grad, grads, state_sums, state_steps): member in Adagrad |
| D | adam.py | 114 def _init_group( member in Adam |
| D | adamw.py | 109 def _init_group( member in AdamW |