Searched refs:kWeightDecay (Results 1 – 2 of 2) sorted by relevance
32 void SGD::set_weight_decay(const float weight_decay) { (void)AddAttr(kWeightDecay, MakeValue(weight… in set_weight_decay()42 auto value_ptr = GetAttr(kWeightDecay); in get_weight_decay()
209 constexpr auto kWeightDecay = "weight_decay"; variable