/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | ForeachFunctors.cuh | 208 using opmath_t = at::opmath_type<T>; typedef 232 using opmath_t = at::opmath_type<T>; typedef 256 using opmath_t = at::opmath_type<T>; typedef 308 using opmath_t = at::opmath_type<T>; typedef 406 using opmath_t = at::opmath_type<T>; typedef 458 using opmath_t = at::opmath_type<T>; typedef 482 using opmath_t = at::opmath_type<T>; typedef 506 using opmath_t = at::opmath_type<T>; typedef 557 using opmath_t = at::opmath_type<T>; typedef 611 using opmath_t = at::opmath_type<T>; typedef
|
H A D | ForeachTernaryOp.cu | 48 using opmath_t = typename at::opmath_type<scalar_t>; in foreach_tensor_lerp_ternary_cuda() typedef 79 using opmath_t = typename at::opmath_type<scalar_t>; in foreach_tensor_lerp_ternary_cuda_() typedef 115 using opmath_t = typename at::opmath_type<scalar_t>; in foreach_tensor_lerp_list_cuda() typedef 147 using opmath_t = typename at::opmath_type<scalar_t>; in foreach_tensor_lerp_list_cuda_() typedef
|
H A D | ForeachPointwiseOp.cu | 48 using opmath_t = at::opmath_type<scalar_t>; in foreach_pointwise_op() typedef 80 using opmath_t = at::opmath_type<scalar_t>; in foreach_pointwise_op_() typedef 112 using opmath_t = at::opmath_type<scalar_t>; in foreach_pointwise_op_() typedef 151 using opmath_t = at::opmath_type<scalar_t>; in foreach_pointwise_op() typedef
|
H A D | ActivationGluKernel.cu | 27 using opmath_t = at::opmath_type<scalar_t>; in glu_kernel() typedef 44 using opmath_t = at::opmath_type<scalar_t>; in glu_jvp_kernel() typedef 84 using opmath_t = at::opmath_type<scalar_t>; in glu_backward_kernel() typedef
|
H A D | PowKernel.cu | 45 using opmath_t = at::opmath_type<scalar_t>; in pow_scalar_tensor_impl() typedef 81 using opmath_t = at::opmath_type<scalar_t>; in pow_chalf_tensor_scalar_impl() typedef 112 using opmath_t = at::opmath_type<scalar_t>; in pow_tensor_tensor_kernel() typedef
|
H A D | ForeachBinaryOpList.cu | 43 using opmath_t = at::opmath_type<T>; in foreach_tensor_list_op() typedef 66 using opmath_t = at::opmath_type<T>; in foreach_tensor_list_op_() typedef 420 using opmath_t = at::opmath_type<scalar_t>; in foreach_tensor_copy_list_kernel_cuda_() typedef
|
H A D | LogAddExpKernel.cu | 20 using opmath_t = at::opmath_type<scalar_t>; in logaddexp_kernel_cuda() typedef 39 using opmath_t = at::opmath_type<scalar_t>; in logaddexp2_kernel_cuda() typedef
|
H A D | BinaryMulKernel.cu | 32 using opmath_t = at::opmath_type<scalar_t>; in mul_kernel_cuda() typedef 39 using opmath_t = at::opmath_type<scalar_t>; in mul_kernel_cuda() typedef
|
H A D | ActivationLogSigmoidKernel.cu | 28 using opmath_t = at::opmath_type<scalar_t>; in launch_log_sigmoid_forward_kernel() typedef 46 using opmath_t = at::opmath_type<scalar_t>; in log_sigmoid_backward_kernel() typedef
|
H A D | BinaryDivTrueKernel.cu | 30 using opmath_t = at::opmath_type<scalar_t>; in div_true_kernel_cuda() typedef 41 using opmath_t = at::opmath_type<scalar_t>; in div_true_kernel_cuda() typedef
|
H A D | ActivationLeakyReluKernel.cu | 29 using opmath_t = at::opmath_type<scalar_t>; in leaky_relu_kernel() typedef 47 using opmath_t = at::opmath_type<scalar_t>; in leaky_relu_backward_kernel() typedef
|
H A D | ActivationHardsigmoidKernel.cu | 29 using opmath_t = at::opmath_type<scalar_t>; in hardsigmoid_kernel() typedef 51 using opmath_t = at::opmath_type<scalar_t>; in hardsigmoid_backward_kernel() typedef
|
H A D | ActivationHardswishKernel.cu | 24 using opmath_t = at::opmath_type<scalar_t>; in hardswish_kernel() typedef 38 using opmath_t = at::opmath_type<scalar_t>; in hardswish_backward_kernel() typedef
|
H A D | ActivationSoftplusKernel.cu | 32 using opmath_t = at::opmath_type<scalar_t>; in softplus_kernel() typedef 54 using opmath_t = at::opmath_type<scalar_t>; in softplus_backward_kernel() typedef
|
H A D | Lerp.cu | 81 using opmath_t = at::opmath_type<scalar_t>; in lerp_scalar_kernel() typedef 112 using opmath_t = at::opmath_type<scalar_t>; in lerp_scalar_kernel() typedef
|
H A D | ActivationEluKernel.cu | 33 using opmath_t = at::opmath_type<scalar_t>; in elu_kernel() typedef 59 using opmath_t = at::opmath_type<scalar_t>; in elu_backward_kernel() typedef
|
H A D | UnarySignKernels.cu | 84 using opmath_t = at::opmath_type<scalar_t>; in signbit_kernel_cuda() typedef 123 using opmath_t = at::opmath_type<scalar_t>; in sgn_kernel_cuda() typedef
|
H A D | ForeachBinaryOpScalarTensor.cu | 48 using opmath_t = at::opmath_type<T>; in foreach_binary_op() typedef 83 using opmath_t = at::opmath_type<T>; in foreach_binary_op_() typedef
|
H A D | AmpKernels.cu | 62 using opmath_t = at::opmath_type<scalar_t>; in _amp_non_finite_check_and_unscale_cuda_() typedef 157 using opmath_t = at::opmath_type<scalar_t>; in _amp_foreach_non_finite_check_and_unscale_cuda_() typedef
|
H A D | ForeachBinaryOpScalarList.cu | 38 using opmath_t = at::opmath_type<T>; in foreach_binary_op() typedef 57 using opmath_t = at::opmath_type<T>; in foreach_binary_op_() typedef
|
H A D | ForeachBinaryOpScalar.cu | 38 using opmath_t = at::opmath_type<T>; in foreach_binary_op() typedef 56 using opmath_t = at::opmath_type<T>; in foreach_binary_op_() typedef
|
H A D | ForeachUnaryOp.cu | 58 using opmath_t = typename at::opmath_type<scalar_t>; in foreach_unary_op() typedef 75 using opmath_t = typename at::opmath_type<scalar_t>; in foreach_unary_op_() typedef
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
H A D | ScatterGatherKernel.cpp | 36 using opmath_t = at::opmath_type<scalar_t>; in operator ()() typedef 50 using opmath_t = at::opmath_type<scalar_t>; in operator ()() typedef 60 using opmath_t = at::opmath_type<scalar_t>; in operator ()() typedef 70 using opmath_t = at::opmath_type<scalar_t>; in operator ()() typedef 80 using opmath_t = at::opmath_type<scalar_t>; in operator ()() typedef 90 using opmath_t = at::opmath_type<scalar_t>; in operator ()() typedef 209 using opmath_t = at::opmath_type<scalar_t>; in operator ()() typedef 299 using opmath_t = at::opmath_type<scalar_t>; in operator ()() typedef 395 using opmath_t = at::opmath_type<scalar_t>; in operator ()() typedef 490 using opmath_t = at::opmath_type<scalar_t>; in operator ()() typedef [all …]
|
H A D | UnaryOpsKernel.cpp | 197 using opmath_t = at::opmath_type<scalar_t>; in abs_kernel() typedef 327 using opmath_t = at::opmath_type<scalar_t>; in signbit_kernel() typedef 337 using opmath_t = at::opmath_type<scalar_t>; in sgn_kernel() typedef 358 using opmath_t = at::opmath_type<scalar_t>; in sinc_kernel() typedef 523 using opmath_t = at::opmath_type<scalar_t>; in kaiser_window_kernel() typedef 626 using opmath_t = at::opmath_type<scalar_t>; in round_decimals_kernel() typedef
|
H A D | UpSampleMoreKernel.cpp | 120 using opmath_t = at::opmath_type<scalar_t>; in cpu_upsample_nearest_backward() typedef 255 using opmath_t = at::opmath_type<scalar_t>; in cpu_upsample_nearest_backward_channels_last() typedef 450 using opmath_t = at::opmath_type<scalar_t>; in cpu_upsample_linear_backward() typedef 625 using opmath_t = at::opmath_type<scalar_t>; in cpu_upsample_linear_backward_channels_last() typedef
|