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