Home
last modified time | relevance | path

Searched refs:gpu_kernel_with_scalars (Results 1 – 25 of 25) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DBinaryRemainderKernel.cu19 gpu_kernel_with_scalars(iter, []GPU_LAMBDA(scalar_t a, scalar_t b) -> scalar_t { in remainder_kernel_cuda()
29 gpu_kernel_with_scalars(iter, in remainder_kernel_cuda()
44 gpu_kernel_with_scalars(iter, []GPU_LAMBDA(scalar_t a, scalar_t b) -> scalar_t { in fmod_kernel_cuda()
50 gpu_kernel_with_scalars(iter, in fmod_kernel_cuda()
H A DStepKernel.cu16 gpu_kernel_with_scalars(iter, []GPU_LAMBDA(scalar_t a, scalar_t b) -> scalar_t { in nextafter_kernel_cuda()
24 gpu_kernel_with_scalars(iter, []GPU_LAMBDA(scalar_t a, scalar_t b) -> scalar_t { in heaviside_kernel_cuda()
H A DBinaryShiftOpsKernels.cu16 gpu_kernel_with_scalars(iter, in lshift_kernel_cuda()
29 gpu_kernel_with_scalars(iter, in rshift_kernel_cuda()
H A DBinaryDivTruncKernel.cu22 gpu_kernel_with_scalars( in div_trunc_kernel_cuda()
42 gpu_kernel_with_scalars( in div_trunc_kernel_cuda()
H A DBinaryDivFloorKernel.cu31 gpu_kernel_with_scalars( in div_floor_kernel_cuda()
72 gpu_kernel_with_scalars( in div_floor_kernel_cuda()
H A DBinaryMiscOpsKernels.cu46 gpu_kernel_with_scalars(iter, []GPU_LAMBDA(scalar_t x, scalar_t y) -> scalar_t { in xlogy_kernel_cuda()
60 gpu_kernel_with_scalars(iter, []GPU_LAMBDA(scalar_t x, scalar_t y) -> scalar_t { in xlog1py_kernel_cuda()
H A DBinaryLogicalOpsKernels.cu71 gpu_kernel_with_scalars(iter, []GPU_LAMBDA(scalar_t a, scalar_t b) -> bool { in logical_or_kernel_cuda()
107 gpu_kernel_with_scalars(iter, []GPU_LAMBDA(scalar_t a, scalar_t b) -> bool { in logical_xor_kernel_cuda()
H A DCopysignKernel.cu25 gpu_kernel_with_scalars(iter, []GPU_LAMBDA(scalar_t a, scalar_t b) -> scalar_t { in copysign_kernel_cuda()
H A Dlegendre_polynomial_p.cu22 … gpu_kernel_with_scalars(iterator, []GPU_LAMBDA(scalar_t x, scalar_t n) -> scalar_t { in legendre_polynomial_p_kernel_cuda()
H A Dshifted_chebyshev_polynomial_u.cu22 … gpu_kernel_with_scalars(iterator, []GPU_LAMBDA(scalar_t x, scalar_t n) -> scalar_t { in shifted_chebyshev_polynomial_u_kernel_cuda()
H A Dchebyshev_polynomial_u.cu22 … gpu_kernel_with_scalars(iterator, []GPU_LAMBDA(scalar_t x, scalar_t n) -> scalar_t { in chebyshev_polynomial_u_kernel_cuda()
H A Dchebyshev_polynomial_t.cu22 … gpu_kernel_with_scalars(iterator, []GPU_LAMBDA(scalar_t x, scalar_t n) -> scalar_t { in chebyshev_polynomial_t_kernel_cuda()
H A Dlaguerre_polynomial_l.cu22 … gpu_kernel_with_scalars(iterator, []GPU_LAMBDA(scalar_t x, scalar_t n) -> scalar_t { in laguerre_polynomial_l_kernel_cuda()
H A Dshifted_chebyshev_polynomial_w.cu22 … gpu_kernel_with_scalars(iterator, []GPU_LAMBDA(scalar_t x, scalar_t n) -> scalar_t { in shifted_chebyshev_polynomial_w_kernel_cuda()
H A Dhermite_polynomial_he.cu22 … gpu_kernel_with_scalars(iterator, []GPU_LAMBDA(scalar_t x, scalar_t n) -> scalar_t { in hermite_polynomial_he_kernel_cuda()
H A Dshifted_chebyshev_polynomial_v.cu22 gpu_kernel_with_scalars(iterator, []GPU_LAMBDA(scalar_t x, scalar_t n) -> scalar_t { in shifted_chebyshev_polynomial_v_kernel_cuda()
H A Dchebyshev_polynomial_w.cu22 … gpu_kernel_with_scalars(iterator, []GPU_LAMBDA(scalar_t x, scalar_t n) -> scalar_t { in chebyshev_polynomial_w_kernel_cuda()
H A Dshifted_chebyshev_polynomial_t.cu22 … gpu_kernel_with_scalars(iterator, []GPU_LAMBDA(scalar_t x, scalar_t n) -> scalar_t { in shifted_chebyshev_polynomial_t_kernel_cuda()
H A DZetaKernel.cu28 gpu_kernel_with_scalars(iter, []GPU_LAMBDA(scalar_t x, scalar_t q) -> scalar_t { in zeta_kernel_cuda()
H A Dhermite_polynomial_h.cu22 … gpu_kernel_with_scalars(iterator, []GPU_LAMBDA(scalar_t x, scalar_t n) -> scalar_t { in hermite_polynomial_h_kernel_cuda()
H A Dchebyshev_polynomial_v.cu22 … gpu_kernel_with_scalars(iterator, []GPU_LAMBDA(scalar_t x, scalar_t n) -> scalar_t { in chebyshev_polynomial_v_kernel_cuda()
H A DActivationThresholdKernel.cu27 gpu_kernel_with_scalars( in threshold_kernel_impl()
H A DBinaryGeometricKernels.cu18 gpu_kernel_with_scalars(iter, []GPU_LAMBDA(scalar_t a, scalar_t b) -> scalar_t { in atan2_kernel_cuda()
H A DBinaryDivTrueKernel.cu53 gpu_kernel_with_scalars(iter, f); in div_true_kernel_cuda()
H A DLoops.cuh239 void gpu_kernel_with_scalars(TensorIteratorBase& iter, const func_t& f) { in gpu_kernel_with_scalars() function