Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/
H A DBatchNorm_miopen.cpp26 bool training, double exponential_average_factor, double epsilon) { in miopen_batch_norm()
62 bool training, double exponential_average_factor, double epsilon) in miopen_batch_norm()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/
H A DBatchNorm.cpp26 double exponential_average_factor, in cudnn_batch_norm()
128 double exponential_average_factor, in cudnn_batch_norm()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/
H A Dbatchnorm.h108 double exponential_average_factor; in forward() local
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc1186 double epsilon, double exponential_average_factor) { in SetBatchNormForwardArgs()
3644 const double exponential_average_factor, in DoBatchNormalizationForward()
3664 const double exponential_average_factor, in DoBatchNormalizationForward()
3686 const double exponential_average_factor, in DoBatchNormalizationForwardImpl()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h1169 const double exponential_average_factor, in DoBatchNormalizationForward()
1189 const double exponential_average_factor, in DoBatchNormalizationForward()
H A Dstream.cc335 const double exponential_average_factor, in ThenBatchNormalizationForward()
389 const double exponential_average_factor, in ThenBatchNormalizationForward()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc5415 const double exponential_average_factor, in DoBatchNormalizationForward()
5439 const double exponential_average_factor, in DoBatchNormalizationForward()
5464 const double exponential_average_factor, in DoBatchNormalizationForwardImpl()