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