Home
last modified time | relevance | path

Searched refs:linear_ (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/tests/ut/cpp/kernel/cpu/
Dsparse_apply_ftrl_cpu_kernel_test.cc38 linear_.clear(); in SetUp()
54 inputs_.push_back(CreateKernelAddress(linear_.data())); in CreateInputAddress()
69 std::vector<float> linear_; member in mindspore::kernel::SparseApplyFtrlCpuKernelTest
81 linear_.push_back(1.0); in TEST_F()
106 linear_.push_back(1.0); in TEST_F()
139 linear_.push_back(1.0); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dsparse_apply_ftrl_cpu_kernel.cc32 auto linear = input_params->linear_; in ComputeFtrl()
166 input_params.linear_ = linear; in LaunchKernel()
Dsparse_optimizer_cpu_kernel.h51 float *linear_{nullptr};