Searched refs:SparseAdamOptimInfoBuilder (Results 1 – 3 of 3) sorted by relevance
64 class SparseAdamOptimInfoBuilder : public OptimizerInfoBuilder {66 explicit SparseAdamOptimInfoBuilder(size_t worker_num) : OptimizerInfoBuilder(worker_num) {} in SparseAdamOptimInfoBuilder() function67 ~SparseAdamOptimInfoBuilder() = default;
146 OptimizerInfo *SparseAdamOptimInfoBuilder::BuildInputs(const WeightPtr &weight, const Keys &, const… in BuildInputs()
74 std::make_shared<SparseAdamOptimInfoBuilder>(worker_num_); in InitOptimInfoBuilders()