Searched refs:FusedBatchNormInferenceFunctor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.h | 49 struct FusedBatchNormInferenceFunctor { struct
|
| D | fused_batch_norm_op.cu.cc | 318 struct FusedBatchNormInferenceFunctor<GPUDevice, T, U> { struct
399 template struct FusedBatchNormInferenceFunctor<GPUDevice, float, float>; variable
400 template struct FusedBatchNormInferenceFunctor<GPUDevice, Eigen::half, float>; variable
|
| D | fused_batch_norm_op.cc | 835 FusedBatchNormInferenceFunctor<GPUDevice, T, U> inference_functor; in operator ()()
1160 void FusedBatchNormInferenceFunctor<GPUDevice, T, U>::operator()( \
1169 extern template struct FusedBatchNormInferenceFunctor<GPUDevice, T, U>;
|