Home
last modified time | relevance | path

Searched defs:scale_offset_desc (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cc907 se::dnn::BatchDescriptor scale_offset_desc; in operator ()() local
1102 se::dnn::BatchDescriptor scale_offset_desc; in operator ()() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddnn.h1168 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward()
1188 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward()
1222 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward()
1240 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward()
Dstream.cc334 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationForward()
360 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationBackward()
388 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationForward()
415 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3643 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward()
3663 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward()
3685 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForwardImpl()
3732 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward()
3750 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward()
3768 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackwardImpl()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc5414 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward()
5438 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward()
5463 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForwardImpl()
5610 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward()
5631 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward()
5653 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackwardImpl()