Searched defs:bias_correction1 (Results 1 – 4 of 4) sorted by relevance
37 const opmath_t& bias_correction1, in adam_math()134 const auto bias_correction1 = 1 - at::native::pow_(beta1, *step_count); in operator ()() local
25 double bias_correction1, in adam_math()168 double bias_correction1, in adam_math()290 double bias_correction1 = 1 - std::pow(beta1, step); in adam_fused_step_impl() local
120 auto bias_correction1 = 1 - std::pow(beta1, state.step()); in step() local
115 auto bias_correction1 = 1 - std::pow(beta1, state.step()); in step() local