Home
last modified time | relevance | path

Searched refs:ReduceSparseGradientParam (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/tests/ut/cpp/kernel/cpu/
Dsparse_optimizer_cpu_kernel_test.cc58 ReduceSparseGradientParam<int> param; in TEST_F()
110 ReduceSparseGradientParam<int> param; in TEST_F()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/
Dcommon_utils_test.cc58 ReduceSparseGradientParam param; in TEST_F()
110 ReduceSparseGradientParam param; in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dsparse_apply_proximal_adagrad_cpu_kernel.cc144 ReduceSparseGradientParam<T> param; in LaunchKernel()
Dsparse_apply_lazy_adam_cpu_kernel.cc147 ReduceSparseGradientParam<T> param; in LaunchKernel()
Dsparse_apply_ftrl_cpu_kernel.cc155 ReduceSparseGradientParam<T> param; in LaunchKernel()
Dsparse_optimizer_cpu_kernel.h38 struct ReduceSparseGradientParam { struct
96 static void BucketReduceSparseGradient(const ReduceSparseGradientParam<T> &param) { in BucketReduceSparseGradient()
Dsparse_apply_adam_cpu_kernel.cc168 ReduceSparseGradientParam<T> param; in LaunchKernel()
/third_party/mindspore/mindspore/ccsrc/ps/
Dutil.cc121 mindspore::kernel::ReduceSparseGradientParam<int> param; in ReduceSparseGradient()