Home
last modified time | relevance | path

Searched refs:est_variance_tensor (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op.cc689 const Tensor& est_variance_tensor = MklGetInput(context, kVarianceIndex); in Compute() local
717 context, est_variance_tensor.dims() == 1, in Compute()
719 est_variance_tensor.shape().DebugString())); in Compute()
812 SetMeanVariance(est_mean_tensor, est_variance_tensor); in Compute()
884 auto est_variance_data = est_variance_tensor.flat<U>().data(); in Compute()