Searched refs:batch_variance_tensor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 742 Tensor* batch_variance_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()
882 auto batch_variance_data = batch_variance_tensor->flat<U>().data(); in Compute()
954 Tensor* batch_variance_tensor = nullptr; in HandleEmptyInput() local
959 &batch_mean_tensor, &batch_variance_tensor, in HandleEmptyInput()
967 Tensor** batch_variance_tensor, in AllocateTFOutputs() argument
972 DCHECK(batch_variance_tensor); in AllocateTFOutputs()
999 batch_variance_tensor, tf_shape_scale, in AllocateTFOutputs()
[all …]
|