/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | cudnn_batchnorm_runner.cc | 34 se::dnn::BatchDescriptor scale_offset_desc; member 67 se::dnn::BatchDescriptor scale_offset_desc; member 104 batch_descs.scale_offset_desc.set_layout(se::dnn::DataLayout::kBatchDepthYX) in MakeBatchNormDescriptors() 126 params->scale_offset_desc = batch_descs.scale_offset_desc; in AssignCommonParams() 144 params->common.scale_offset_desc, // in RunCudnnBatchNormForwardInferenceImpl() 173 params->common.scale_offset_desc, // in RunCudnnBatchNormForwardTrainingImpl() 201 params->common.scale_offset_desc, // in RunCudnnBatchNormBackwardImpl()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 224 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, 238 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, 251 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, 262 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, 595 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, 609 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
|
D | cuda_dnn.cc | 3453 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward() argument 3464 scale_offset_desc, epsilon, exponential_average_factor, in DoBatchNormalizationForward() 3476 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward() argument 3487 scale_offset_desc, epsilon, exponential_average_factor, in DoBatchNormalizationForward() 3501 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForwardImpl() argument 3510 scale_offset_desc, ToCudnnDataType(scale_data_type)); in DoBatchNormalizationForwardImpl() 3647 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward() argument 3654 x, scale, mean, inv_var, x_desc, scale_offset_desc, in DoBatchNormalizationBackward() 3665 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward() argument 3672 x, scale, mean, inv_var, x_desc, scale_offset_desc, in DoBatchNormalizationBackward() [all …]
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 262 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, 276 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, 289 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, 300 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, 711 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, 724 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon,
|
D | rocm_dnn.cc | 3527 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward() argument 3536 estimated_mean, estimated_variance, side_input, x_desc, scale_offset_desc, in DoBatchNormalizationForward() 3547 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward() argument 3556 estimated_mean, estimated_variance, side_input, x_desc, scale_offset_desc, in DoBatchNormalizationForward() 3569 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForwardImpl() argument 3580 scale_offset_desc, ToMIOpenDataType(scale_data_type)}; in DoBatchNormalizationForwardImpl() 3615 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward() argument 3622 x_desc, scale_offset_desc, epsilon, x_backprop, scale_backprop, in DoBatchNormalizationBackward() 3631 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward() argument 3638 x_desc, scale_offset_desc, epsilon, x_backprop, scale_backprop, in DoBatchNormalizationBackward() [all …]
|
/external/tensorflow/tensorflow/core/kernels/ |
D | fused_batch_norm_op.cc | 897 se::dnn::BatchDescriptor scale_offset_desc; in operator ()() local 898 scale_offset_desc.set_count(1) in operator ()() 959 scale_offset_desc, static_cast<double>(epsilon), in operator ()() 1088 se::dnn::BatchDescriptor scale_offset_desc; in operator ()() local 1089 scale_offset_desc.set_count(1) in operator ()() 1130 x_desc, scale_offset_desc, static_cast<double>(epsilon), in operator ()()
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 1017 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward() argument 1036 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationForward() argument 1069 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward() argument 1085 const dnn::BatchDescriptor& scale_offset_desc, const double epsilon, in DoBatchNormalizationBackward() argument
|
D | stream.cc | 348 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationForward() argument 357 PARAM(scale_offset_desc), PARAM(epsilon), PARAM(y)); in ThenBatchNormalizationForward() 361 x_desc, scale_offset_desc, epsilon, exponential_average_factor, in ThenBatchNormalizationForward() 374 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationBackward() argument 380 PARAM(scale_offset_desc), PARAM(epsilon), PARAM(x_backprop), in ThenBatchNormalizationBackward() 384 this, y_backprop, x, scale, mean, inv_var, x_desc, scale_offset_desc, in ThenBatchNormalizationBackward() 399 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationForward() argument 408 PARAM(scale_offset_desc), PARAM(epsilon), PARAM(y)); in ThenBatchNormalizationForward() 412 x_desc, scale_offset_desc, epsilon, exponential_average_factor, in ThenBatchNormalizationForward() 426 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, in ThenBatchNormalizationBackward() argument [all …]
|
D | stream.h | 252 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, 265 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, 277 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon, 291 const dnn::BatchDescriptor &scale_offset_desc, const double epsilon,
|