Searched defs:dgamma (Results 1 – 5 of 5) sorted by relevance
63 … float *dgamma, float *dbeta, const float *ws_dgamma, const float *ws_dbeta) { in ComputeMeanKernel()83 float *dgamma, float *dbeta, const float *ws_dgamma, const float *ws_dbeta, in ComputeMean()
27 float dgamma = 0.0f; in LayerNormFp16Grad() local
30 float dgamma = 0.0f; in LayerNormGrad() local
72 auto dgamma = GetDeviceAddress<float>(outputs, 1); in Launch() local
98 T dgamma = (T)0.0; in LaunchKernel() local