Searched refs:saved_inv_variance (Results 1 – 1 of 1) sorted by relevance
1016 void* saved_inv_variance, in SetBatchNormForwardArgs() argument1030 saved_inv_variance, running_mean, running_variance, exp_avg_factor, in SetBatchNormForwardArgs()1044 const void* saved_inv_variance) { in SetBatchNormBackwardArgs() argument1055 offset_grad, saved_mean, saved_inv_variance); in SetBatchNormBackwardArgs()