Searched refs:est_mean_tensor (Results 1 – 1 of 1) sorted by relevance
688 const Tensor& est_mean_tensor = MklGetInput(context, kMeanIndex); in Compute() local713 context, est_mean_tensor.dims() == 1, in Compute()715 est_mean_tensor.shape().DebugString())); in Compute()812 SetMeanVariance(est_mean_tensor, est_variance_tensor); in Compute()883 auto est_mean_data = est_mean_tensor.flat<U>().data(); in Compute()