Searched refs:batch_mean_tensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 741 Tensor* batch_mean_tensor = nullptr; in Compute() local
792 &batch_mean_tensor, &batch_variance_tensor, in Compute()
804 &batch_mean_tensor, &batch_variance_tensor, in Compute()
810 SetMeanVariance(*batch_mean_tensor, *batch_variance_tensor); in Compute()
881 auto batch_mean_data = batch_mean_tensor->flat<U>().data(); in Compute()
953 Tensor* batch_mean_tensor = nullptr; in HandleEmptyInput() local
959 &batch_mean_tensor, &batch_variance_tensor, in HandleEmptyInput()
966 Tensor** batch_mean_tensor, in AllocateTFOutputs() argument
971 DCHECK(batch_mean_tensor); in AllocateTFOutputs()
985 AllocateOutputSetMklShape(context, kBatchMeanIndex, batch_mean_tensor, in AllocateTFOutputs()
[all …]
|