| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | FusedAdam.cpp | 33 const std::optional<at::Tensor>& grad_scale, in _fused_adam_kernel_cpu_()
90 const std::optional<at::Tensor>& grad_scale, in _fused_adam_kernel_cpu_()
109 const std::optional<at::Tensor>& grad_scale, in _fused_adamw_kernel_cpu_()
166 const std::optional<at::Tensor>& grad_scale, in _fused_adamw_kernel_cpu_()
|
| H A D | FusedSGD.cpp | 29 const std::optional<at::Tensor>& grad_scale, in _fused_sgd_kernel_cpu_()
74 const std::optional<at::Tensor>& grad_scale, in _fused_sgd_kernel_cpu_()
|
| H A D | FusedAdagrad.cpp | 27 const std::optional<at::Tensor>& grad_scale, in _fused_adagrad_kernel_cpu_()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | FusedSgdKernel.cu | 156 const std::optional<at::Tensor>& grad_scale, in _fused_sgd_with_momentum_kernel_cuda_()
202 const std::optional<at::Tensor>& grad_scale, in _fused_sgd_with_momentum_kernel_cuda_()
278 const std::optional<at::Tensor>& grad_scale, in _fused_sgd_kernel_cuda_()
342 const std::optional<at::Tensor>& grad_scale, in _fused_sgd_kernel_cuda_()
|
| H A D | FusedAdamKernel.cu | 30 const std::optional<at::Tensor>& grad_scale, in _fused_adam_kernel_cuda_()
89 const std::optional<at::Tensor>& grad_scale, in _fused_adam_kernel_cuda_()
|
| H A D | FusedAdamWKernel.cu | 31 const std::optional<at::Tensor>& grad_scale, in _fused_adamw_kernel_cuda_()
90 const std::optional<at::Tensor>& grad_scale, in _fused_adamw_kernel_cuda_()
|
| H A D | fused_adam_impl.cu | 23 const std::optional<at::Tensor>& grad_scale, in _fused_adam_cuda_impl_()
69 const std::optional<at::Tensor>& grad_scale, in _fused_adam_cuda_impl_()
|
| H A D | fused_adamw_impl.cu | 24 const std::optional<at::Tensor>& grad_scale, in _fused_adamw_cuda_impl_()
70 const std::optional<at::Tensor>& grad_scale, in _fused_adamw_cuda_impl_()
|
| H A D | fused_adam_amsgrad_impl.cu | 24 const std::optional<at::Tensor>& grad_scale, in _fused_adam_amsgrad_cuda_impl_()
75 const std::optional<at::Tensor>& grad_scale, in _fused_adam_amsgrad_cuda_impl_()
|
| H A D | fused_adamw_amsgrad_impl.cu | 25 const std::optional<at::Tensor>& grad_scale, in _fused_adamw_amsgrad_cuda_impl_()
76 const std::optional<at::Tensor>& grad_scale, in _fused_adamw_amsgrad_cuda_impl_()
|
| H A D | Normalization.cuh | 396 stat_accscalar_t grad_scale = invstd * weight_val; in batch_norm_backward_kernel() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | batchnorm_expander.cc | 493 auto grad_scale = add_binary(feature_shape, HloOpcode::kMultiply, in HandleBatchNormGrad() local
|
| /aosp_15_r20/external/pytorch/test/cpp_extensions/ |
| H A D | open_registration_extension.cpp | 638 at::Tensor grad_scale = at::empty({}).fill_(0.00001).to(at::DeviceType::PrivateUse1); in fallback_with_undefined_tensor() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | training_ops.cc | 1006 xla::XlaOp grad_scale = ComputeGradientScale(alpha, decay); in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | batch_normalization_test.cc | 811 auto grad_scale = ReferenceUtil::Reduce4DTo1D( in XLA_TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | training_ops_gpu.cu.cc | 1072 auto grad_scale = (logbase_bcast * sign_decay_bcast * sign_gm).exp(); in operator ()() local
|
| H A D | training_ops.cc | 953 auto grad_scale = (logbase() * sign_decay() * sign_gm).exp(); in operator ()() local
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 3169 grad_scale=None, argument
3195 grad_scale=None, argument
|