Home
last modified time | relevance | path

Searched defs:param_state (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/train/optim/
H A Dsgd.cpp103 auto param_state = state_.find(p.unsafeGetTensorImpl()); in step() local
/aosp_15_r20/external/pytorch/torch/csrc/api/src/optim/
H A Dsgd.cpp86 auto param_state = state_.find(p.unsafeGetTensorImpl()); in step() local
H A Drmsprop.cpp88 auto param_state = state_.find(p.unsafeGetTensorImpl()); in step() local
H A Dadam.cpp87 auto param_state = state_.find(p.unsafeGetTensorImpl()); in step() local
H A Dadamw.cpp87 auto param_state = state_.find(p.unsafeGetTensorImpl()); in step() local
H A Dlbfgs.cpp436 auto param_state = state_.find(_params.at(0).unsafeGetTensorImpl()); in step() local
/aosp_15_r20/external/executorch/extension/training/optimizer/
H A Dsgd.cpp107 auto param_state = state_.find(p.unsafeGetTensorImpl()); in step() local
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/optim/
H A Dserialize.h42 DerivedOptimizerParamState param_state; in serialize() local