Searched defs:param_groups (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/optim/ |
| H A D | optimizer.cpp | 164 std::vector<OptimizerParamGroup>& Optimizer::param_groups() noexcept { in param_groups() function in torch::optim::Optimizer
168 const std::vector<OptimizerParamGroup>& Optimizer::param_groups() in param_groups() function in torch::optim::Optimizer
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/optim/ |
| H A D | serialize.h | 53 const std::vector<OptimizerParamGroup>& param_groups) { in serialize()
88 std::pair<std::vector<std::string>, std::unique_ptr<OptimizerOptions>>>& in serialize()
|
| H A D | lbfgs.h | 64 std::move(param_groups), in Optimizer()
|
| H A D | rmsprop.h | 65 std::move(param_groups), in Optimizer()
|
| H A D | sgd.h | 56 std::vector<OptimizerParamGroup> param_groups, in SGD()
|
| H A D | adam.h | 60 std::move(param_groups), in Optimizer()
|
| H A D | adamw.h | 60 std::move(param_groups), in Optimizer()
|
| H A D | adagrad.h | 65 std::move(param_groups), in Optimizer()
|
| H A D | optimizer.h | 118 std::vector<OptimizerParamGroup> param_groups, in Optimizer()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/train/optim/ |
| H A D | sgd.h | 77 const std::vector<torch::jit::mobile::SGDParamGroup>& param_groups, in SGD()
|
| /aosp_15_r20/external/executorch/extension/training/optimizer/ |
| H A D | sgd.h | 183 const std::vector<SGDParamGroup>& param_groups, in SGD()
|
| /aosp_15_r20/external/pytorch/torch/optim/ |
| H A D | optimizer.py | 872 def _cast(param, value, param_id=None, param_groups=None, key=None): argument
|