| /external/executorch/kernels/test/ |
| D | op_native_batch_norm_test.cpp | 29 double momentum, in op_native_batch_norm_legit_no_training_out()
59 double momentum, in op_native_batch_norm_legit_out()
89 double momentum, in op_native_batch_norm_legit_no_stats_out()
160 double momentum = 0.1; in TEST_F() local
283 double momentum = 0.1; in TEST_F() local
452 double momentum = 0.1; in TEST_F() local
612 double momentum = 0.1; in TEST_F() local
753 double momentum = 0.1; in TEST_F() local
856 double momentum = 0.1; in TEST_F() local
945 double momentum = 0.1; in TEST_F() local
[all …]
|
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | Normalization.cpp | 27 double momentum, in mkldnn_batch_norm()
51 double momentum, in _mkldnn_batch_norm_legit()
60 double momentum, in _mkldnn_batch_norm_legit_no_stats()
67 Tensor& running_mean, Tensor& running_var, double momentum, double eps) { in _batch_norm_with_update_mkldnn()
136 double momentum, in mkldnn_batch_norm()
213 Tensor& running_mean, Tensor& running_var, double momentum, double eps) { in _batch_norm_with_update_mkldnn()
224 double momentum, in _mkldnn_batch_norm_legit()
233 double momentum, in _mkldnn_batch_norm_legit_no_stats()
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | FusedSgdKernel.cu | 17 const double momentum, in sgd_math()
150 const double momentum, in _fused_sgd_with_momentum_kernel_cuda_()
196 const double momentum, in _fused_sgd_with_momentum_kernel_cuda_()
272 const double momentum, in _fused_sgd_kernel_cuda_()
336 const double momentum, in _fused_sgd_kernel_cuda_()
|
| D | Normalization.cu | 367 const auto momentum = static_cast<acc_t>(momentum_); in batch_norm_update_stats() local
403 const auto momentum = static_cast<acc_t>(momentum_); in batch_norm_update_stats_and_invert() local
435 …_mean_opt, const std::optional<Tensor>& running_var_opt, bool train, double momentum, double epsil… in batch_norm_cuda_out()
461 …_mean_opt, const std::optional<Tensor>& running_var_opt, bool train, double momentum, double epsil… in batch_norm_cuda()
486 Tensor& running_mean, Tensor& running_var, double momentum, double eps) { in _batch_norm_with_update_cuda()
511 Tensor& running_mean, Tensor& running_var, double momentum, double eps, in _batch_norm_with_update_cuda_out()
532 …r>& bias_opt, Tensor& running_mean, Tensor& running_var, bool train, double momentum, double epsil… in _batch_norm_legit_cuda()
536 …sor>& weight_opt, const std::optional<Tensor>& bias_opt, bool train, double momentum, double epsil… in _batch_norm_legit_no_stats_cuda()
540 …r>& bias_opt, Tensor& running_mean, Tensor& running_var, bool train, double momentum, double epsil… in _batch_norm_legit_cuda_out()
544 …sor>& weight_opt, const std::optional<Tensor>& bias_opt, bool train, double momentum, double epsil… in _batch_norm_legit_no_stats_cuda_out()
[all …]
|
| /external/tensorflow/tensorflow/python/training/ |
| D | rmsprop_test.py | 57 def _rmsprop_update_numpy(self, var, g, mg, rms, mom, lr, decay, momentum, argument
71 lr, decay, momentum, epsilon, centered): argument
469 momentum = lambda: 0.0 function
|
| D | momentum_test.py | 36 def _update_nesterov_momentum_numpy(self, var, accum, g, lr, momentum): argument
58 momentum = lambda: 0.9 function
|
| D | momentum.py | 116 def __init__(self, learning_rate, momentum, argument
|
| D | rmsprop.py | 139 momentum=0.0, argument
|
| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | FusedSGDKernel.cpp | 23 const double momentum, in sgd_math()
115 const double momentum, in sgd_math()
190 const double momentum, in sgd_fused_step_impl()
240 const double momentum, in fused_sgd_kernel()
|
| /external/pytorch/torch/ao/nn/quantized/modules/ |
| D | batchnorm.py | 11 self, num_features, eps=1e-5, momentum=0.1, device=None, dtype=None argument
57 self, num_features, eps=1e-5, momentum=0.1, device=None, dtype=None argument
97 def __init__(self, num_features, eps=1e-5, momentum=0.1, device=None, dtype=None): argument
|
| D | normalization.py | 165 momentum=0.1, argument
232 momentum=0.1, argument
299 momentum=0.1, argument
|
| /external/pytorch/aten/src/ATen/native/ |
| D | Normalization.cpp | 202 double momentum, double eps, Tensor& save_mean, Tensor& save_var_transform) { in batch_norm_cpu_update_stats_template()
286 double momentum, double eps) { in batch_norm_cpu_update_stats_template()
542 bool training, double momentum, double eps, bool cudnn_enabled) { in _batch_norm_impl_index()
671 bool training, double momentum, double eps, bool cudnn_enabled) { in batch_norm()
706 bool use_input_stats, double momentum, double eps, bool cudnn_enabled) { in instance_norm()
743 …ptional<Tensor>& running_mean_opt, const std::optional<Tensor>& running_var_opt, double momentum) { in batch_norm_update_stats_cpu()
762 … bool train, double momentum, double eps, Tensor& out, Tensor& save_mean, Tensor& save_var) { in batch_norm_cpu_out()
805 bool train, double momentum, double eps) { in batch_norm_cpu()
855 Tensor& running_mean, Tensor& running_var, double momentum, double eps) { in _batch_norm_with_update_cpu()
864 Tensor& running_mean, Tensor& running_var, double momentum, double eps, in _batch_norm_with_update_cpu_out()
[all …]
|
| D | FusedSGD.cpp | 23 const double momentum, in _fused_sgd_kernel_cpu_()
68 const double momentum, in _fused_sgd_kernel_cpu_()
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | optimization_parameters.proto | 103 float momentum = 1; field
208 float momentum = 1; field
220 float momentum = 2; field
232 float momentum = 2; field
542 MomentumParameters momentum = 8; field
|
| /external/executorch/kernels/portable/cpu/ |
| D | op_native_batch_norm.cpp | 31 double momentum, in _native_batch_norm_legit_no_training_out()
147 double momentum, in _native_batch_norm_legit_out()
183 double momentum, in _native_batch_norm_legit_no_stats_out()
|
| /external/pytorch/torch/ao/nn/intrinsic/quantized/modules/ |
| D | bn_relu.py | 24 def __init__(self, num_features, eps=1e-5, momentum=0.1, device=None, dtype=None): argument
72 def __init__(self, num_features, eps=1e-5, momentum=0.1, device=None, dtype=None): argument
|
| /external/pytorch/torch/csrc/api/include/torch/nn/functional/ |
| D | instancenorm.h | 18 double momentum, in instance_norm()
|
| D | batchnorm.h | 20 std::optional<double> momentum, in batch_norm()
|
| /external/pytorch/torch/ao/nn/intrinsic/qat/modules/ |
| D | conv_fused.py | 60 momentum=0.1, argument
474 momentum=0.1, argument
545 momentum=0.1, argument
671 momentum=0.1, argument
742 momentum=0.1, argument
868 momentum=0.1, argument
938 momentum=0.1, argument
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | momentum_test.py | 31 def _update_nesterov_momentum_numpy(self, var, accum, g, lr, momentum): argument
|
| D | rmsprop_test.py | 37 momentum=0.0, argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | single_loss_example.py | 80 momentum=0.9, argument
|
| /external/pytorch/test/quantization/eager/ |
| D | test_quantize_eager_qat.py | 79 eps=1e-05, momentum=0.1, argument
240 eps=1e-05, momentum=0.1, argument
776 momentum, argument
927 momentum, argument
|
| /external/pytorch/torch/nn/modules/ |
| D | _functions.py | 17 momentum, argument
|
| /external/executorch/extension/training/optimizer/ |
| D | sgd.h | 103 double momentum() const { in momentum() function
|