Searched refs:batch_variance_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_fused_batch_norm_op.cc | 803 Tensor* batch_variance_tensor = nullptr; in Compute() local
853 &batch_mean_tensor, &batch_variance_tensor, in Compute()
865 &batch_mean_tensor, &batch_variance_tensor, in Compute()
871 SetMeanVariance(*batch_mean_tensor, *batch_variance_tensor); in Compute()
944 auto batch_variance_data = batch_variance_tensor->flat<U>().data(); in Compute()
1016 Tensor* batch_variance_tensor = nullptr; in HandleEmptyInput() local
1021 &batch_mean_tensor, &batch_variance_tensor, in HandleEmptyInput()
1029 Tensor** batch_variance_tensor, in AllocateTFOutputs() argument
1034 DCHECK(batch_variance_tensor); in AllocateTFOutputs()
1061 batch_variance_tensor, tf_shape_scale, in AllocateTFOutputs()
[all …]
|