Searched refs:RunCudnnBatchNormBackwardImpl (Results 1 – 1 of 1) sorted by relevance
190 void RunCudnnBatchNormBackwardImpl(CudnnBatchNormBackwardParams* params, in RunCudnnBatchNormBackwardImpl() function302 RunCudnnBatchNormBackwardImpl<Eigen::half>(&backward_params, stream); in RunCudnnBatchNormBackward()305 RunCudnnBatchNormBackwardImpl<float>(&backward_params, stream); in RunCudnnBatchNormBackward()