Searched defs:lr_ (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ | ||
| D | replace_momentum_cast_fusion.h | 40 VarPtr lr_; variable |
| D | adam_fusion.h | 49 VarPtr lr_; variable |
| D | adam_weight_decay_fusion.h | 50 VarPtr lr_; variable |
| /third_party/mindspore/tests/ut/cpp/kernel/cpu/ | ||
| D | adam_delta_cpu_kernel_test.cc | 71 float lr_ = 0.001; member in mindspore::kernel::AdamDeltaCpuKernelTest |
| D | sparse_apply_proximal_adagrad_cpu_kernel_test.cc | 72 float lr_ = 0.01; member in mindspore::kernel::SparseApplyProximalAdagradCpuKernelTest |
| D | sparse_apply_lazy_adam_cpu_kernel_test.cc | 79 float lr_ = 0.001; member in mindspore::kernel::SparseApplyLazyAdamCpuKernelTest |
| D | sparse_apply_adam_cpu_kernel_test.cc | 81 float lr_ = 0.001; member in mindspore::kernel::SparseApplyAdamCpuKernelTest |
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ | ||
| D | sparse_ftrl_gpu_kernel.h | 165 float lr_; variable |
| /third_party/mindspore/mindspore/lite/src/train/ | ||
| D | optimizer_kernel.h | 214 float lr_ = 0.0f; variable |