Searched defs:bias_correction2 (Results 1 – 4 of 4) sorted by relevance
26 double bias_correction2, in adam_math()169 double bias_correction2, in adam_math()291 double bias_correction2 = 1 - std::pow(beta2, step); in adam_fused_step_impl() local
121 auto bias_correction2 = 1 - std::pow(beta2, state.step()); in step() local
116 auto bias_correction2 = 1 - std::pow(beta2, state.step()); in step() local
135 const auto bias_correction2 = 1 - at::native::pow_(beta2, *step_count); in operator ()() local