Searched defs:max_exp_avg_sq_ptr (Results 1 – 1 of 1) sorted by relevance
23 scalar_t* max_exp_avg_sq_ptr, in adam_math()166 scalar_t* max_exp_avg_sq_ptr, in adam_math()309 scalar_t* max_exp_avg_sq_ptr = amsgrad ? max_exp_avg_sq_data + begin : nullptr; in adam_fused_step_impl() local