Searched refs:sparse_ftrl_ (Results 1 – 1 of 1) sorted by relevance
29 SparseApplyFtrlCpuKernelTest() : sparse_ftrl_(std::make_shared<SparseApplyFtrlCPUKernel>()) {} in SparseApplyFtrlCpuKernelTest()32 sparse_ftrl_->lr_ = 0.001; in SetUp()33 sparse_ftrl_->l1_ = 0.0; in SetUp()34 sparse_ftrl_->l2_ = 0.0; in SetUp()35 sparse_ftrl_->lr_power_ = -0.5; in SetUp()74 std::shared_ptr<SparseApplyFtrlCPUKernel> sparse_ftrl_; member in mindspore::kernel::SparseApplyFtrlCpuKernelTest84 sparse_ftrl_->indices_size_ = 3; in TEST_F()85 sparse_ftrl_->var_first_dim_size_ = 3; in TEST_F()86 sparse_ftrl_->var_outer_dim_size_ = 9; in TEST_F()87 sparse_ftrl_->indices_data_type_ = kNumberTypeInt64; in TEST_F()[all …]