Home
last modified time | relevance | path

Searched refs:scale_offset_desc (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_runner.cc34 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/
Dcuda_dnn.h224 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,
Dcuda_dnn.cc3453 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/
Drocm_dnn.h262 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,
Drocm_dnn.cc3527 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/
Dfused_batch_norm_op.cc897 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/
Ddnn.h1017 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
Dstream.cc348 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 …]
Dstream.h252 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,