Searched refs:SetMeanVariance (Results 1 – 1 of 1) sorted by relevance
810 SetMeanVariance(*batch_mean_tensor, *batch_variance_tensor); in Compute()812 SetMeanVariance(est_mean_tensor, est_variance_tensor); in Compute()933 void SetMeanVariance(const Tensor& mean, const Tensor& variance) { in SetMeanVariance() function in tensorflow::MklFusedBatchNormOp