Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.h279 bool DoBatchNormalizationForward(
293 bool DoBatchNormalizationForward(
H A Dcuda_dnn.cc5408 bool CudnnSupport::DoBatchNormalizationForward( in DoBatchNormalizationForward() function in stream_executor::gpu::CudnnSupport
5431 bool CudnnSupport::DoBatchNormalizationForward( in DoBatchNormalizationForward() function in stream_executor::gpu::CudnnSupport
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.h283 bool DoBatchNormalizationForward(
297 bool DoBatchNormalizationForward(
H A Drocm_dnn.cc3636 bool MIOpenSupport::DoBatchNormalizationForward( in DoBatchNormalizationForward() function in stream_executor::gpu::MIOpenSupport
3657 bool MIOpenSupport::DoBatchNormalizationForward( in DoBatchNormalizationForward() function in stream_executor::gpu::MIOpenSupport
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h1162 virtual bool DoBatchNormalizationForward( in DoBatchNormalizationForward() function
1181 virtual bool DoBatchNormalizationForward( in DoBatchNormalizationForward() function
H A Dstream.cc344 CheckError(dnn->DoBatchNormalizationForward( in ThenBatchNormalizationForward()
398 CheckError(dnn->DoBatchNormalizationForward( in ThenBatchNormalizationForward()