Searched refs:inference_functor (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 835 FusedBatchNormInferenceFunctor<GPUDevice, T, U> inference_functor; in operator ()() local
838 inference_functor(context, tensor_format, x.tensor<T, 4>(), in operator ()()
845 inference_functor(context, tensor_format, x.tensor<T, 4>(), in operator ()()
|