Searched defs:scale_offset_desc (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_thunk.cc | 68 dnn::BatchDescriptor scale_offset_desc; in MakeDescriptors() local
105 dnn::BatchDescriptor scale_offset_desc; in ExecuteOnStream() local
167 dnn::BatchDescriptor scale_offset_desc; in ExecuteOnStream() local
251 dnn::BatchDescriptor scale_offset_desc; in ExecuteOnStream() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 310 se::dnn::BatchDescriptor scale_offset_desc; in operator ()() local
464 se::dnn::BatchDescriptor scale_offset_desc; in operator ()() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3184 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward()
3205 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward()
3228 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForwardImpl()
3284 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward()
3299 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward()
3315 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3002 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward()
3021 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward()
3042 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForwardImpl()
3089 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward()
3103 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward()
3118 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 345 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationForward()
371 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationBackward()
395 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationForward()
422 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationBackward()
|