Searched defs:momentum_ (Results 1 – 3 of 3) sorted by relevance
121 double momentum_; variable
349 double momentum_, int64_t N) { in batch_norm_update_stats()383 double momentum_, double epsilon, int64_t N) { in batch_norm_update_stats_and_invert()
225 auto momentum_ = static_cast<param_t>(momentum); in batch_norm_cpu_update_stats_template() local