Home
last modified time | relevance | path

Searched refs:RunCudnnBatchNormForwardInferenceImpl (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_runner.cc133 void RunCudnnBatchNormForwardInferenceImpl( in RunCudnnBatchNormForwardInferenceImpl() function
240 RunCudnnBatchNormForwardInferenceImpl<Eigen::half>(&inference_params, in RunCudnnBatchNormForwardInference()
244 RunCudnnBatchNormForwardInferenceImpl<float>(&inference_params, stream); in RunCudnnBatchNormForwardInference()