Home
last modified time | relevance | path

Searched refs:MOMENTUM (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dbatchnorm_info.cc37 momentum_ = GetFloatAttr(MOMENTUM); in GetAttrs()
227 Attr attr_momentum = std::make_pair(MOMENTUM, momentum); in InferReplaceOps()
Dops_utils.h226 constexpr char MOMENTUM[] = "momentum"; variable
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dsgd_cpu_kernel.h42 enum input_list_ { PARAM, GRAD, LR, ACCUM, MOMENTUM, STAT }; enumerator
Dsgd_cpu_kernel.cc45 auto momentum = reinterpret_cast<T *>(inputs[MOMENTUM]->addr); in Launch()
Dcpu_kernel.h84 constexpr char MOMENTUM[] = "momentum"; variable