Home
last modified time | relevance | path

Searched defs:opmath_t (Results 1 – 25 of 58) sorted by relevance

123

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DForeachFunctors.cuh208 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 DForeachTernaryOp.cu48 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 DForeachPointwiseOp.cu48 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 DActivationGluKernel.cu27 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 DPowKernel.cu45 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 DForeachBinaryOpList.cu43 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 DLogAddExpKernel.cu20 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 DBinaryMulKernel.cu32 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 DActivationLogSigmoidKernel.cu28 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 DBinaryDivTrueKernel.cu30 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 DActivationLeakyReluKernel.cu29 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 DActivationHardsigmoidKernel.cu29 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 DActivationHardswishKernel.cu24 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 DActivationSoftplusKernel.cu32 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 DLerp.cu81 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 DActivationEluKernel.cu33 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 DUnarySignKernels.cu84 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 DForeachBinaryOpScalarTensor.cu48 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 DAmpKernels.cu62 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 DForeachBinaryOpScalarList.cu38 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 DForeachBinaryOpScalar.cu38 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 DForeachUnaryOp.cu58 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 DScatterGatherKernel.cpp36 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 DUnaryOpsKernel.cpp197 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 DUpSampleMoreKernel.cpp120 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

123