BucketReduceSparseGradient (reference) in projects: third_party

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

/third_party/mindspore/tests/ut/cpp/kernel/cpu/
D	sparse_optimizer_cpu_kernel_test.cc	`64 SparseOptimizerCPUKernel::BucketReduceSparseGradient(param); in TEST_F() 116 SparseOptimizerCPUKernel::BucketReduceSparseGradient(param); in TEST_F()`
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/
D	common_utils_test.cc	`64 BucketReduceSparseGradient(param); in TEST_F() 116 BucketReduceSparseGradient(param); in TEST_F()`
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
D	sparse_apply_proximal_adagrad_cpu_kernel.cc	`150 BucketReduceSparseGradient(param); in LaunchKernel()`
D	sparse_apply_lazy_adam_cpu_kernel.cc	`153 BucketReduceSparseGradient(param); in LaunchKernel()`
D	sparse_apply_ftrl_cpu_kernel.cc	`161 BucketReduceSparseGradient(param); in LaunchKernel()`
D	sparse_apply_adam_cpu_kernel.cc	`174 BucketReduceSparseGradient(param); in LaunchKernel()`
D	sparse_optimizer_cpu_kernel.h	`96 static void BucketReduceSparseGradient(const ReduceSparseGradientParam<T> &param) { in BucketReduceSparseGradient() function`
/third_party/mindspore/mindspore/ccsrc/ps/
D	util.cc	`128 mindspore::kernel::SparseOptimizerCPUKernel::BucketReduceSparseGradient(param); in ReduceSparseGradient()`