Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h557 port::Status DoBatchNormalizationForwardImpl(
Dcuda_dnn.cc3191 DoBatchNormalizationForwardImpl<float, float>( in DoBatchNormalizationForward()
3212 DoBatchNormalizationForwardImpl<Eigen::half, float>( in DoBatchNormalizationForward()
3221 port::Status CudnnSupport::DoBatchNormalizationForwardImpl( in DoBatchNormalizationForwardImpl() function in stream_executor::gpu::CudnnSupport
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h629 bool DoBatchNormalizationForwardImpl(
Drocm_dnn.cc3008 return DoBatchNormalizationForwardImpl<Eigen::half, float>( in DoBatchNormalizationForward()
3027 return DoBatchNormalizationForwardImpl<float, float>( in DoBatchNormalizationForward()
3035 bool MIOpenSupport::DoBatchNormalizationForwardImpl( in DoBatchNormalizationForwardImpl() function in stream_executor::gpu::MIOpenSupport