Searched refs:saved_var (Results 1 – 2 of 2) sorted by relevance
88 typename TTypes<U>::Vec saved_var(saved_var_output->vec<U>()); in operator ()() local132 saved_var.device(d) = variance; in operator ()()
1384 void* saved_mean, void* saved_var, in SetBatchNormForwardArgs() argument1390 saved_mean, saved_var, epsilon); in SetBatchNormForwardArgs()1485 const void* saved_var, in SetBatchNormBackwardArgs() argument1492 offset_grad, saved_mean, saved_var); in SetBatchNormBackwardArgs()