Home
last modified time | relevance | path

Searched refs:exponential_average_factor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h225 const double exponential_average_factor,
239 const double exponential_average_factor,
596 const double exponential_average_factor,
Dcuda_dnn.cc3454 const double exponential_average_factor, in DoBatchNormalizationForward() argument
3464 scale_offset_desc, epsilon, exponential_average_factor, in DoBatchNormalizationForward()
3477 const double exponential_average_factor, in DoBatchNormalizationForward() argument
3487 scale_offset_desc, epsilon, exponential_average_factor, in DoBatchNormalizationForward()
3502 const double exponential_average_factor, in DoBatchNormalizationForwardImpl() argument
3578 if (exponential_average_factor == 1.0) { in DoBatchNormalizationForwardImpl()
3608 /*exponentialAverageFactor=*/exponential_average_factor, in DoBatchNormalizationForwardImpl()
3626 scale.opaque(), offset.opaque(), exponential_average_factor, in DoBatchNormalizationForwardImpl()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h263 const double exponential_average_factor,
277 const double exponential_average_factor,
712 const double exponential_average_factor,
Drocm_dnn.cc1132 double epsilon, double exponential_average_factor) { in SetBatchNormForwardArgs() argument
1144 exponential_average_factor); in SetBatchNormForwardArgs()
3528 const double exponential_average_factor, in DoBatchNormalizationForward() argument
3537 epsilon, exponential_average_factor, activation_mode, y, batch_mean, in DoBatchNormalizationForward()
3548 const double exponential_average_factor, in DoBatchNormalizationForward() argument
3557 epsilon, exponential_average_factor, activation_mode, y, batch_mean, in DoBatchNormalizationForward()
3570 const double exponential_average_factor, in DoBatchNormalizationForwardImpl() argument
3591 exponential_average_factor, batch_mean->opaque(), batch_var->opaque(), in DoBatchNormalizationForwardImpl()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1018 const double exponential_average_factor, in DoBatchNormalizationForward() argument
1037 const double exponential_average_factor, in DoBatchNormalizationForward() argument
Dstream.h253 const double exponential_average_factor,
278 const double exponential_average_factor,
Dstream.cc349 const double exponential_average_factor, in ThenBatchNormalizationForward() argument
361 x_desc, scale_offset_desc, epsilon, exponential_average_factor, in ThenBatchNormalizationForward()
400 const double exponential_average_factor, in ThenBatchNormalizationForward() argument
412 x_desc, scale_offset_desc, epsilon, exponential_average_factor, in ThenBatchNormalizationForward()