Searched defs:gradient_scale (Results 1 – 3 of 3) sorted by relevance
28 … ApplyMomentum::Init(const bool use_nesterov, const bool use_locking, const float gradient_scale) { in Init()42 void ApplyMomentum::set_gradient_scale(const float gradient_scale) { in set_gradient_scale()
51 uint8_t gradient_scale[3]; member
2658 def __init__(self, use_nesterov=False, use_locking=False, gradient_scale=1.0): argument