Searched refs:kSparseLazyAdam (Results 1 – 3 of 3) sorted by relevance
52 constexpr char kSparseLazyAdam[] = "LazyAdam"; variable241 … {kSparseLazyAdam, kSparseAdamOriginIdx},245 … {kSparseLazyAdam, kSparseAdamPSSendIdx},
30 {kSparseLazyAdam, 2},37 {2, kSparseLazyAdam},
128 } else if (optim_name == kSparseLazyAdam) { in InitOptimInputsShape()