Searched refs:grad_mean (Results 1 – 1 of 1) sorted by relevance
492 def _FoldFusedBatchNormGrad(op, unused_grad_y, grad_mean, grad_var, unused_1, argument496 array_ops.size(x) / array_ops.size(grad_mean), dtypes.float32)497 dmean_dx = grad_mean / n