/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | Math.cuh | 14 const auto ndtri_string = jiterator_stringify( 177 const auto log_ndtr_string = jiterator_stringify( 190 const auto gcd_string = jiterator_stringify( 206 const auto lcm_string = jiterator_stringify( 232 const auto digamma_string = jiterator_stringify( 300 const auto zeta_string = jiterator_stringify( 386 const auto trigamma_string = jiterator_stringify( 412 const auto lgamma_string = jiterator_stringify( 419 const auto polygamma_string = zeta_string + jiterator_stringify( 429 const auto exp2_string = jiterator_stringify( [all …]
|
H A D | ReduceSumProdKernel.cu | 32 std::string func = jiterator_stringify( in operator ()() 65 std::string func = jiterator_stringify( in operator ()() 89 std::string func = jiterator_stringify( in operator ()() 127 jiterator_stringify(arg_t combine(arg_t a, arg_t b) { return a * b; }); in operator ()()
|
H A D | BinaryLogicalOpsKernels.cu | 19 static const auto logical_and_string = jiterator_stringify( in logical_and_kernel_cuda() 56 static const auto logical_or_string = jiterator_stringify( in logical_or_kernel_cuda() 92 static const auto logical_xor_string = jiterator_stringify( in logical_xor_kernel_cuda()
|
H A D | UnaryLogKernels.cu | 23 static const auto log_string = jiterator_stringify( in log_kernel_cuda() 55 static const auto log10_string = jiterator_stringify( in log10_kernel_cuda() 92 static const auto log2_string = jiterator_stringify( in log2_kernel_cuda()
|
H A D | UnaryOpsKernel.cu | 42 static const auto exp_string = jiterator_stringify( in exp_kernel_cuda() 100 static const auto rsqrt_string = jiterator_stringify( in rsqrt_kernel_cuda() 139 static const auto sqrt_string = jiterator_stringify( in sqrt_kernel_cuda()
|
H A D | UnaryComplexKernels.cu | 36 static const auto angle_string = jiterator_stringify( in angle_kernel_cuda() 71 static const auto conj_string = jiterator_stringify( in conj_kernel_cuda()
|
H A D | UnarySignKernels.cu | 33 static const auto neg_string = jiterator_stringify( in neg_kernel_cuda() 103 static const auto sgn_string = jiterator_stringify( in sgn_kernel_cuda()
|
H A D | Lerp.cu | 17 static const auto lerp_tensor_string = jiterator_stringify( in lerp_tensor_kernel() 71 static const auto lerp_scalar_string = jiterator_stringify( in lerp_scalar_kernel()
|
H A D | BinaryMiscBackwardOpsKernels.cu | 23 static const auto sigmoid_backward_string = jiterator_stringify( in sigmoid_backward_kernel_cuda() 94 static const auto tanh_backward_string = jiterator_stringify( in tanh_backward_kernel_cuda()
|
H A D | PointwiseOpsKernel.cu | 26 static const auto addcmul_string = jiterator_stringify( in addcmul_cuda_kernel() 74 jiterator_stringify(template <typename T> T addcdiv( in addcdiv_cuda_kernel()
|
H A D | PowKernel.cu | 51 jiterator_stringify(template <typename T> T pow_scalar_base_kernel(T exp, T fct) { in pow_scalar_tensor_impl() 73 jiterator_stringify(template <typename T> T pow_kernel(T base, T exp) {
|
H A D | UnaryGeometricAsinKernel.cu | 24 static const auto asin_string = jiterator_stringify( in asin_kernel_cuda()
|
H A D | BinaryMulKernel.cu | 27 static const auto mul_string = jiterator_stringify( in mul_kernel_cuda()
|
H A D | AbsKernel.cu | 23 static const auto abs_string = jiterator_stringify( in abs_kernel_cuda()
|
H A D | UnaryGeometricCoshKernel.cu | 23 static const auto cosh_string = jiterator_stringify( in cosh_kernel_cuda()
|
H A D | UnaryGeometricTanKernel.cu | 23 static const auto tan_string = jiterator_stringify( in tan_kernel_cuda()
|
H A D | UnaryGeometricAtanKernel.cu | 23 static const auto atan_string = jiterator_stringify( in atan_kernel_cuda()
|
H A D | UnaryGeometricSinKernel.cu | 23 static const auto sin_string = jiterator_stringify( in sin_kernel_cuda()
|
H A D | UnaryGeometricTanhKernel.cu | 23 static const auto tanh_string = jiterator_stringify( in tanh_kernel_cuda()
|
H A D | UnaryGeometricCosKernel.cu | 23 static const auto cos_string = jiterator_stringify( in cos_kernel_cuda()
|
H A D | UnaryGeometricAcoshKernel.cu | 24 static const auto acosh_string = jiterator_stringify( in acosh_kernel_cuda()
|
H A D | UnaryGeometricSinhKernel.cu | 23 static const auto sinh_string = jiterator_stringify( in sinh_kernel_cuda()
|
H A D | UnaryGeometricAsinhKernel.cu | 24 static const auto asinh_string = jiterator_stringify( in asinh_kernel_cuda()
|
H A D | UnaryGeometricAtanhKernel.cu | 23 static const auto atanh_string = jiterator_stringify( in atanh_kernel_cuda()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/ |
H A D | jit_macros.h | 7 #define jiterator_stringify(...) std::string(#__VA_ARGS__); macro
|