Searched defs:use_nesterov (Results 1 – 3 of 3) sorted by relevance
117 use_locking=False, name="Momentum", use_nesterov=False): argument
106 bool use_nesterov = 2; field209 bool use_nesterov = 2; field
185 bool use_nesterov) { in ApplyAdamKernel()224 const T* momentum, bool use_nesterov, Tindex param_rows, in SparseApplyKerasMomentumKernel()