Home
last modified time | relevance | path

Searched defs:self_val (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DLerpKernel.cpp71 [weight_val](BFloat16 self_val, BFloat16 end_val) -> BFloat16 { in lerp_scalar_kernel()
88 [weight_val](Half self_val, Half end_val) -> Half { in lerp_scalar_kernel()
103 [weight_val](scalar_t self_val, scalar_t end_val) { in lerp_scalar_kernel()
119 [=](BFloat16 self_val, BFloat16 end_val, BFloat16 weight_val) -> BFloat16 { in lerp_tensor_kernel()
134 [=](Half self_val, Half end_val, Half weight_val) -> Half { in lerp_tensor_kernel()
149 [](scalar_t self_val, scalar_t end_val, scalar_t weight_val) { in lerp_tensor_kernel()
152 … [](Vectorized<scalar_t> self_val, Vectorized<scalar_t> end_val, Vectorized<scalar_t> weight_val) { in lerp_tensor_kernel()
H A DActivation.cpp629 [&](scalar_t self_val) -> scalar_t { in hardsigmoid_kernel()
632 [&](vec::Vectorized<scalar_t> self_val) -> vec::Vectorized<scalar_t> { in hardsigmoid_kernel()
656 [&](scalar_t self_val) { in hardsigmoid_kernel()
659 [&](Vec self_val) { in hardsigmoid_kernel()
681 [=](scalar_t grad_val, scalar_t self_val) -> scalar_t { in hardsigmoid_backward_kernel()
686 [=](Vectorized<scalar_t> grad_val, Vectorized<scalar_t> self_val) -> Vectorized<scalar_t> { in hardsigmoid_backward_kernel()
707 [=](scalar_t grad_val, scalar_t self_val) { in hardsigmoid_backward_kernel()
712 [=](Vec grad_val, Vec self_val) { in hardsigmoid_backward_kernel()
726 [=](scalar_t self_val) { in hardshrink_kernel()
730 [=](Vec self_val) { in hardshrink_kernel()
[all …]
H A DPointwiseOpsKernel.cpp20 [=](scalar_t self_val, scalar_t t1_val, scalar_t t2_val) -> scalar_t { in addcmul_cpu_kernel()
41 [=](scalar_t self_val, scalar_t t1_val, scalar_t t2_val) -> scalar_t { in addcmul_cpu_kernel()
61 [=](scalar_t self_val, scalar_t t1_val, scalar_t t2_val) -> scalar_t { in addcdiv_cpu_kernel()
81 [=](scalar_t self_val, scalar_t t1_val, scalar_t t2_val) -> scalar_t { in addcdiv_cpu_kernel()
225 [=](scalar_t self_val, scalar_t t1_val, scalar_t t2_val) -> scalar_t { in mse_backward_cpu_kernel()
H A DLinearAlgebraKernel.cpp34 scalar_t vec2_val) __ubsan_ignore_undefined__ -> scalar_t { in addr_kernel()
72 scalar_t vec2_val) __ubsan_ignore_undefined__ -> scalar_t { in addr_kernel()
H A DTensorCompareKernel.cpp220 [=](bool cond_val, scalar_t self_val, scalar_t other_val) -> scalar_t { in where_kernel_impl()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DDistributions.cpp399 cpu_serial_kernel(iter, [](scalar_t self_val, scalar_t output_val) -> scalar_t{ in _standard_gamma_grad_cpu()