| /external/pytorch/torch/csrc/api/include/torch/nn/modules/ |
| D | batchnorm.h | 108 double exponential_average_factor; in forward() local
110 exponential_average_factor = 0.0; in forward()
112 exponential_average_factor = this->options.momentum().value(); in forward()
120 exponential_average_factor = in forward()
123 exponential_average_factor = this->options.momentum().value(); in forward()
135 /*momentum=*/exponential_average_factor, in forward()
|
| /external/pytorch/torch/nn/modules/ |
| D | batchnorm.py | 166 exponential_average_factor = 0.0
168 exponential_average_factor = self.momentum
175 exponential_average_factor = 1.0 / float(self.num_batches_tracked)
177 exponential_average_factor = self.momentum
203 exponential_average_factor,
738 exponential_average_factor = 0.0
740 exponential_average_factor = self.momentum
746 exponential_average_factor = 1.0 / self.num_batches_tracked.item()
748 exponential_average_factor = self.momentum
805 exponential_average_factor,
[all …]
|
| /external/pytorch/aten/src/ATen/native/miopen/ |
| D | BatchNorm_miopen.cpp | 26 bool training, double exponential_average_factor, double epsilon) { in miopen_batch_norm() argument
62 bool training, double exponential_average_factor, double epsilon) in miopen_batch_norm() argument
129 exponential_average_factor, in miopen_batch_norm()
|
| /external/pytorch/torch/utils/ |
| D | mkldnn.py | 145 self.exponential_average_factor = 0.0
147 self.exponential_average_factor = dense_module.momentum
180 self.exponential_average_factor,
|
| /external/pytorch/aten/src/ATen/native/cudnn/ |
| D | BatchNorm.cpp | 26 double exponential_average_factor, in cudnn_batch_norm() argument
128 double exponential_average_factor, in cudnn_batch_norm() argument
228 exponential_average_factor, in cudnn_batch_norm()
|
| /external/pytorch/test/quantization/eager/ |
| D | test_quantize_eager_qat.py | 142 exponential_average_factor = 0.0
144 exponential_average_factor = self.momentum
151 exponential_average_factor = 1.0 / float(self.num_batches_tracked)
153 exponential_average_factor = self.momentum
181 self.running_mean = exponential_average_factor * batch_mean.detach() + \
182 (1 - exponential_average_factor) * self.running_mean
183 self.running_var = exponential_average_factor * unbiased_batch_var.detach() + \
184 (1 - exponential_average_factor) * self.running_var
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.h | 286 const double exponential_average_factor,
301 const double exponential_average_factor,
545 const double exponential_average_factor,
|
| D | cuda_dnn.cc | 5415 const double exponential_average_factor, in DoBatchNormalizationForward() argument
5425 scale_offset_desc, epsilon, exponential_average_factor, in DoBatchNormalizationForward()
5439 const double exponential_average_factor, in DoBatchNormalizationForward() argument
5449 scale_offset_desc, epsilon, exponential_average_factor, in DoBatchNormalizationForward()
5464 const double exponential_average_factor, in DoBatchNormalizationForwardImpl() argument
5540 if (exponential_average_factor == 1.0) { in DoBatchNormalizationForwardImpl()
5570 /*exponentialAverageFactor=*/exponential_average_factor, in DoBatchNormalizationForwardImpl()
5588 scale.opaque(), offset.opaque(), exponential_average_factor, in DoBatchNormalizationForwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 290 const double exponential_average_factor,
305 const double exponential_average_factor,
649 const double exponential_average_factor,
|
| D | rocm_dnn.cc | 1186 double epsilon, double exponential_average_factor) { in SetBatchNormForwardArgs() argument
1198 exponential_average_factor); in SetBatchNormForwardArgs()
3644 const double exponential_average_factor, in DoBatchNormalizationForward() argument
3653 epsilon, exponential_average_factor, activation_mode, y, batch_mean, in DoBatchNormalizationForward()
3664 const double exponential_average_factor, in DoBatchNormalizationForward() argument
3673 epsilon, exponential_average_factor, activation_mode, y, batch_mean, in DoBatchNormalizationForward()
3686 const double exponential_average_factor, in DoBatchNormalizationForwardImpl() argument
3707 exponential_average_factor, batch_mean->opaque(), batch_var->opaque(), in DoBatchNormalizationForwardImpl()
|
| /external/pytorch/torch/_inductor/ |
| D | decomposition.py | 760 exponential_average_factor: float,
770 exponential_average_factor,
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream.cc | 335 const double exponential_average_factor, in ThenBatchNormalizationForward() argument
346 x_desc, scale_offset_desc, epsilon, exponential_average_factor, in ThenBatchNormalizationForward()
389 const double exponential_average_factor, in ThenBatchNormalizationForward() argument
400 x_desc, scale_offset_desc, epsilon, exponential_average_factor, in ThenBatchNormalizationForward()
|
| D | dnn.h | 1169 const double exponential_average_factor, in DoBatchNormalizationForward() argument
1189 const double exponential_average_factor, in DoBatchNormalizationForward() argument
|
| D | stream.h | 259 const double exponential_average_factor,
286 const double exponential_average_factor,
|
| /external/pytorch/test/inductor/ |
| D | test_torchinductor.py | 3146 exponential_average_factor = 0.0
3148 exponential_average_factor = self.momentum
3155 exponential_average_factor = 1.0 / float(
3159 exponential_average_factor = self.momentum
3182 exponential_average_factor,
|
| /external/pytorch/torch/_decomp/ |
| D | decompositions.py | 2212 exponential_average_factor: float,
2222 exponential_average_factor,
|
| /external/pytorch/torch/ |
| D | overrides.py | 787 …mbda input, weight, bias, running_mean, running_var, training, exponential_average_factor, epsilon…
|