Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h572 port::Status DoBatchNormalizationBackwardImpl(
Dcuda_dnn.cc3287 return IsStatusOk(DoBatchNormalizationBackwardImpl( in DoBatchNormalizationBackward()
3302 return IsStatusOk(DoBatchNormalizationBackwardImpl( in DoBatchNormalizationBackward()
3310 port::Status CudnnSupport::DoBatchNormalizationBackwardImpl( in DoBatchNormalizationBackwardImpl() function in stream_executor::gpu::CudnnSupport
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h644 bool DoBatchNormalizationBackwardImpl(
Drocm_dnn.cc3092 return DoBatchNormalizationBackwardImpl<Eigen::half, float>( in DoBatchNormalizationBackward()
3106 return DoBatchNormalizationBackwardImpl<float, float>( in DoBatchNormalizationBackward()
3113 bool MIOpenSupport::DoBatchNormalizationBackwardImpl( in DoBatchNormalizationBackwardImpl() function in stream_executor::gpu::MIOpenSupport