Searched refs:saved_mean_tensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 743 Tensor* saved_mean_tensor = nullptr; in Compute() local
793 &saved_mean_tensor, &saved_variance_tensor, in Compute()
805 &saved_mean_tensor, &saved_variance_tensor, in Compute()
824 reinterpret_cast<char*>(saved_mean_tensor->flat<U>().data()); in Compute()
863 U* mean_op_data = saved_mean_tensor->flat<U>().data(); in Compute()
955 Tensor* saved_mean_tensor = nullptr; in HandleEmptyInput() local
960 &saved_mean_tensor, &saved_variance_tensor, in HandleEmptyInput()
968 Tensor** saved_mean_tensor, in AllocateTFOutputs() argument
973 DCHECK(saved_mean_tensor); in AllocateTFOutputs()
1010 AllocateOutputSetMklShape(context, kSavedMeanIndex, saved_mean_tensor, in AllocateTFOutputs()
[all …]
|