Searched refs:opmath_symmetric_gpu_kernel_with_scalars (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | MaxMinElementwiseKernel.cu | 16 opmath_symmetric_gpu_kernel_with_scalars<bool>( in maximum_kernel_cuda()
22 opmath_symmetric_gpu_kernel_with_scalars<scalar_t>( in maximum_kernel_cuda()
29 opmath_symmetric_gpu_kernel_with_scalars<scalar_t>( in maximum_kernel_cuda()
45 opmath_symmetric_gpu_kernel_with_scalars<bool>(iter, []GPU_LAMBDA(bool a, bool b) -> bool { in minimum_kernel_cuda()
50 …opmath_symmetric_gpu_kernel_with_scalars<scalar_t>(iter, []GPU_LAMBDA(scalar_t a, scalar_t b) -> s… in minimum_kernel_cuda()
56 …opmath_symmetric_gpu_kernel_with_scalars<scalar_t>(iter, []GPU_LAMBDA(scalar_t a, scalar_t b) -> s… in minimum_kernel_cuda()
72 …opmath_symmetric_gpu_kernel_with_scalars<scalar_t>(iter, []GPU_LAMBDA(scalar_t a, scalar_t b) -> s… in fmax_kernel_cuda()
84 …opmath_symmetric_gpu_kernel_with_scalars<scalar_t>(iter, []GPU_LAMBDA(scalar_t a, scalar_t b) -> s… in fmin_kernel_cuda()
|
| H A D | BinaryLogicalOpsKernels.cu | 34 opmath_symmetric_gpu_kernel_with_scalars<scalar_t, bool>( in logical_and_kernel_cuda()
43 opmath_symmetric_gpu_kernel_with_scalars<scalar_t, bool>( in logical_and_kernel_cuda()
79 opmath_symmetric_gpu_kernel_with_scalars<scalar_t, bool>( in logical_or_kernel_cuda()
115 opmath_symmetric_gpu_kernel_with_scalars<scalar_t, bool>( in logical_xor_kernel_cuda()
|
| H A D | BinaryBitwiseOpsKernels.cu | 30 opmath_symmetric_gpu_kernel_with_scalars<scalar_t>(iter, f); in bitwise_and_kernel_cuda()
51 opmath_symmetric_gpu_kernel_with_scalars<scalar_t>(iter, f); in bitwise_or_kernel_cuda()
72 opmath_symmetric_gpu_kernel_with_scalars<scalar_t>(iter, f); in bitwise_xor_kernel_cuda()
|
| H A D | BinaryMulKernel.cu | 33 opmath_symmetric_gpu_kernel_with_scalars<scalar_t>( in mul_kernel_cuda()
40 opmath_symmetric_gpu_kernel_with_scalars<scalar_t>( in mul_kernel_cuda()
|
| H A D | BinaryGeometricKernels.cu | 29 opmath_symmetric_gpu_kernel_with_scalars<scalar_t>( in hypot_kernel_cuda()
|
| H A D | CompareEQKernel.cu | 34 opmath_symmetric_gpu_kernel_with_scalars<scalar_t, bool>( in compare_eq_ne_kernel()
|
| H A D | Loops.cuh | 197 void opmath_symmetric_gpu_kernel_with_scalars(TensorIteratorBase& iter, const func_t& f) { in opmath_symmetric_gpu_kernel_with_scalars() function
|