Searched refs:BinaryFuncVariant (Results 1 – 10 of 10) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | JitLoops.cuh | 72 at::cuda::jit::BinaryFuncVariant scalar_pos = in jitted_gpu_kernel()
73 at::cuda::jit::BinaryFuncVariant::NoScalar, in jitted_gpu_kernel()
121 if (scalar_pos == at::cuda::jit::BinaryFuncVariant::NoScalar) { in jitted_gpu_kernel()
130 at::cuda::jit::BinaryFuncVariant::NoScalar>( in jitted_gpu_kernel()
132 } else if (scalar_pos == at::cuda::jit::BinaryFuncVariant::RhsScalar) { in jitted_gpu_kernel()
138 at::cuda::jit::BinaryFuncVariant::RhsScalar>( in jitted_gpu_kernel()
151 at::cuda::jit::BinaryFuncVariant::LhsScalar>( in jitted_gpu_kernel()
175 …kernel<name, return_type, f_inputs_type, 1>(iter, f, at::cuda::jit::BinaryFuncVariant::LhsScalar, … in opmath_jitted_gpu_kernel_with_scalars()
179 …kernel<name, return_type, f_inputs_type, 1>(iter, f, at::cuda::jit::BinaryFuncVariant::RhsScalar, … in opmath_jitted_gpu_kernel_with_scalars()
|
| D | CUDAJitLoops.cuh | 87 at::cuda::jit::BinaryFuncVariant scalar_pos, in launch_jitted_unrolled_kernel()
115 at::cuda::jit::BinaryFuncVariant scalar_pos, in launch_jitted_vectorized_kernel()
180 at::cuda::jit::BinaryFuncVariant scalar_pos, in jitted_gpu_kernel_generic()
260 at::cuda::jit::BinaryFuncVariant scalar_pos =
261 at::cuda::jit::BinaryFuncVariant::NoScalar,
|
| D | jit_utils.h | 14 enum class BinaryFuncVariant {NoScalar, RhsScalar, LhsScalar}; enum
95 BinaryFuncVariant scalar_pos,
105 BinaryFuncVariant scalar_pos,
|
| D | PointwiseOpsKernel.cu | 35 /*scalar_pos=*/at::cuda::jit::BinaryFuncVariant::NoScalar, in addcmul_cuda_kernel()
83 /*scalar_pos=*/at::cuda::jit::BinaryFuncVariant::NoScalar, in addcdiv_cuda_kernel()
|
| D | PowKernel.cu | 57 /*scalar_pos=*/at::cuda::jit::BinaryFuncVariant::NoScalar, in pow_scalar_tensor_impl()
87 /*scalar_pos=*/at::cuda::jit::BinaryFuncVariant::NoScalar, in pow_chalf_tensor_scalar_impl()
|
| D | UnaryGammaKernels.cu | 87 /*scalar_pos=*/at::cuda::jit::BinaryFuncVariant::NoScalar, in polygamma_kernel_cuda()
|
| D | jit_utils.cpp | 940 BinaryFuncVariant scalar_pos, in generate_code()
980 BinaryFuncVariant scalar_pos, in generate_code()
1029 if (scalar_pos == BinaryFuncVariant::NoScalar) { in generate_code()
1034 } else if (scalar_pos == BinaryFuncVariant::LhsScalar) { in generate_code()
|
| D | Lerp.cu | 90 /*scalar_pos=*/ at::cuda::jit::BinaryFuncVariant::NoScalar, in lerp_scalar_kernel()
|
| D | UnarySpecialOpsKernel.cu | 332 /*scalar_pos=*/at::cuda::jit::BinaryFuncVariant::NoScalar, in kaiser_window_kernel_cuda()
|
| /external/pytorch/aten/src/ATen/cuda/ |
| D | jiterator.cu | 43 ss << static_cast<int>(at::cuda::jit::BinaryFuncVariant::NoScalar); in launch_jitted_vectorized_kernel_dynamic()
61 at::cuda::jit::BinaryFuncVariant::NoScalar, in launch_jitted_vectorized_kernel_dynamic()
141 ss << static_cast<int>(at::cuda::jit::BinaryFuncVariant::NoScalar); in launch_jitted_unrolled_kernel_dynamic()
156 at::cuda::jit::BinaryFuncVariant::NoScalar, in launch_jitted_unrolled_kernel_dynamic()
|