Home
last modified time | relevance | path

Searched refs:std_vec2 (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/cpu/
DFusedAdagradKernel.cpp57 fVec std_vec2 = state_sum_vec2.sqrt() + fVec(scalar_t(eps)); in adagrad_math() local
59 param_vec2 = param_vec2 - fVec(scalar_t(clr)) * grad_vec2 / std_vec2; in adagrad_math()