| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Activation.cpp | 270 TORCH_IMPL_FUNC(elu_out) ( in TORCH_IMPL_FUNC() function
276 TORCH_IMPL_FUNC(elu_backward_out) ( in TORCH_IMPL_FUNC() function
288 TORCH_IMPL_FUNC(silu_out) ( in TORCH_IMPL_FUNC() function
294 TORCH_IMPL_FUNC(silu_backward_out) ( in TORCH_IMPL_FUNC() function
300 TORCH_IMPL_FUNC(mish_out) ( in TORCH_IMPL_FUNC() function
306 TORCH_IMPL_FUNC(softplus_out) ( in TORCH_IMPL_FUNC() function
312 TORCH_IMPL_FUNC(softplus_backward_out) ( in TORCH_IMPL_FUNC() function
322 TORCH_IMPL_FUNC(leaky_relu_out) ( in TORCH_IMPL_FUNC() function
328 TORCH_IMPL_FUNC(leaky_relu_backward_out) ( in TORCH_IMPL_FUNC() function
338 TORCH_IMPL_FUNC(hardsigmoid_out) ( in TORCH_IMPL_FUNC() function
[all …]
|
| H A D | TensorCompare.cpp | 646 TORCH_IMPL_FUNC(max_out) in TORCH_IMPL_FUNC() function
655 TORCH_IMPL_FUNC(min_out) in TORCH_IMPL_FUNC() function
692 TORCH_IMPL_FUNC(clamp_out) in TORCH_IMPL_FUNC() function
713 TORCH_IMPL_FUNC(clamp_Tensor_out) in TORCH_IMPL_FUNC() function
725 TORCH_IMPL_FUNC(clamp_max_out) in TORCH_IMPL_FUNC() function
737 TORCH_IMPL_FUNC(clamp_max_Tensor_out) in TORCH_IMPL_FUNC() function
742 TORCH_IMPL_FUNC(clamp_min_out) in TORCH_IMPL_FUNC() function
751 TORCH_IMPL_FUNC(clamp_min_Tensor_out) in TORCH_IMPL_FUNC() function
805 TORCH_IMPL_FUNC(isin_Tensor_Tensor_out) ( in TORCH_IMPL_FUNC() function
823 TORCH_IMPL_FUNC(isin_Tensor_Scalar_out) ( in TORCH_IMPL_FUNC() function
[all …]
|
| H A D | BinaryOps.cpp | 433 TORCH_IMPL_FUNC(sub_out) ( in TORCH_IMPL_FUNC() function
440 TORCH_IMPL_FUNC(mul_out) ( in TORCH_IMPL_FUNC() function
446 TORCH_IMPL_FUNC(div_out) (const Tensor& self, const Tensor& other, const Tensor& result) { in TORCH_IMPL_FUNC() function
450 TORCH_IMPL_FUNC(div_out_mode) ( in TORCH_IMPL_FUNC() function
462 TORCH_IMPL_FUNC(logit_backward_out) (const Tensor& grad_output, const Tensor& input, std::optional<… in TORCH_IMPL_FUNC() function
466 TORCH_IMPL_FUNC(sigmoid_backward_out) (const Tensor& grad_output, const Tensor& output, const Tenso… in TORCH_IMPL_FUNC() function
470 TORCH_IMPL_FUNC(special_xlog1py_out) (const Tensor& self, const Tensor& other, const Tensor& result… in TORCH_IMPL_FUNC() function
474 TORCH_IMPL_FUNC(special_zeta_out) (const Tensor& self, const Tensor& other, const Tensor& result) { in TORCH_IMPL_FUNC() function
478 TORCH_IMPL_FUNC(special_chebyshev_polynomial_t_out) (const Tensor& self, const Tensor& n, const Ten… in TORCH_IMPL_FUNC() function
482 TORCH_IMPL_FUNC(special_chebyshev_polynomial_u_out) (const Tensor& self, const Tensor& n, const Ten… in TORCH_IMPL_FUNC() function
[all …]
|
| H A D | UpSampleNearest1d.cpp | 80 TORCH_IMPL_FUNC(upsample_nearest1d_out_cpu) ( in TORCH_IMPL_FUNC() function
89 TORCH_IMPL_FUNC(_upsample_nearest_exact1d_out_cpu) ( in TORCH_IMPL_FUNC() function
98 TORCH_IMPL_FUNC(upsample_nearest1d_backward_out_cpu) ( in TORCH_IMPL_FUNC() function
109 TORCH_IMPL_FUNC(_upsample_nearest_exact1d_backward_out_cpu) ( in TORCH_IMPL_FUNC() function
|
| H A D | UpSampleNearest2d.cpp | 104 TORCH_IMPL_FUNC(upsample_nearest2d_out_cpu) ( in TORCH_IMPL_FUNC() function
114 TORCH_IMPL_FUNC(_upsample_nearest_exact2d_out_cpu) ( in TORCH_IMPL_FUNC() function
124 TORCH_IMPL_FUNC(upsample_nearest2d_backward_out_cpu) ( in TORCH_IMPL_FUNC() function
135 TORCH_IMPL_FUNC(_upsample_nearest_exact2d_backward_out_cpu) ( in TORCH_IMPL_FUNC() function
|
| H A D | UpSampleNearest3d.cpp | 113 TORCH_IMPL_FUNC(upsample_nearest3d_out_cpu) ( in TORCH_IMPL_FUNC() function
124 TORCH_IMPL_FUNC(_upsample_nearest_exact3d_out_cpu) ( in TORCH_IMPL_FUNC() function
135 TORCH_IMPL_FUNC(upsample_nearest3d_backward_out_cpu) ( in TORCH_IMPL_FUNC() function
147 TORCH_IMPL_FUNC(_upsample_nearest_exact3d_backward_out_cpu) ( in TORCH_IMPL_FUNC() function
|
| H A D | UpSampleBilinear2d.cpp | 108 TORCH_IMPL_FUNC(upsample_bilinear2d_out_cpu) ( in TORCH_IMPL_FUNC() function
119 TORCH_IMPL_FUNC(upsample_bilinear2d_backward_out_cpu) ( in TORCH_IMPL_FUNC() function
133 TORCH_IMPL_FUNC(_upsample_bilinear2d_aa_out_cpu) ( in TORCH_IMPL_FUNC() function
144 TORCH_IMPL_FUNC(_upsample_bilinear2d_aa_backward_out_cpu) ( in TORCH_IMPL_FUNC() function
|
| H A D | ReduceOps.cpp | 408 TORCH_IMPL_FUNC(aminmax_out) in TORCH_IMPL_FUNC() function
491 TORCH_IMPL_FUNC(cumsum_out) in TORCH_IMPL_FUNC() function
499 TORCH_IMPL_FUNC(cumprod_out) in TORCH_IMPL_FUNC() function
1197 TORCH_IMPL_FUNC(sum_out) in TORCH_IMPL_FUNC() function
1322 TORCH_IMPL_FUNC(prod_out) in TORCH_IMPL_FUNC() function
1348 TORCH_IMPL_FUNC(mean_out) in TORCH_IMPL_FUNC() function
1534 TORCH_IMPL_FUNC(norm_out) in TORCH_IMPL_FUNC() function
1543 TORCH_IMPL_FUNC(norm_dtype_out) in TORCH_IMPL_FUNC() function
1611 TORCH_IMPL_FUNC(all_out) in TORCH_IMPL_FUNC() function
1616 TORCH_IMPL_FUNC(all_dims_out) in TORCH_IMPL_FUNC() function
[all …]
|
| H A D | UpSampleBicubic2d.cpp | 226 TORCH_IMPL_FUNC(upsample_bicubic2d_out_cpu) ( in TORCH_IMPL_FUNC() function
237 TORCH_IMPL_FUNC(upsample_bicubic2d_backward_out_cpu) ( in TORCH_IMPL_FUNC() function
250 TORCH_IMPL_FUNC(_upsample_bicubic2d_aa_out_cpu) ( in TORCH_IMPL_FUNC() function
261 TORCH_IMPL_FUNC(_upsample_bicubic2d_aa_backward_out_cpu) ( in TORCH_IMPL_FUNC() function
|
| H A D | ReflectionPad.cpp | 312 TORCH_IMPL_FUNC(reflection_pad1d_out_cpu) in TORCH_IMPL_FUNC() function
317 TORCH_IMPL_FUNC(reflection_pad1d_backward_out_cpu)(const Tensor& grad_output, in TORCH_IMPL_FUNC() function
371 TORCH_IMPL_FUNC(reflection_pad3d_out_cpu) in TORCH_IMPL_FUNC() function
379 TORCH_IMPL_FUNC(reflection_pad3d_backward_out_cpu)(const Tensor& grad_output, in TORCH_IMPL_FUNC() function
|
| H A D | ReplicationPadding.cpp | 271 TORCH_IMPL_FUNC(replication_pad1d_out_cpu) ( in TORCH_IMPL_FUNC() function
277 TORCH_IMPL_FUNC(replication_pad1d_backward_out_cpu) ( in TORCH_IMPL_FUNC() function
288 TORCH_IMPL_FUNC(replication_pad2d_out_cpu) ( in TORCH_IMPL_FUNC() function
320 TORCH_IMPL_FUNC(replication_pad3d_out_cpu) ( in TORCH_IMPL_FUNC() function
|
| H A D | Pow.cpp | 47 TORCH_IMPL_FUNC(pow_Tensor_Tensor_out) (const Tensor& base, const Tensor& exp, const Tensor& out) { in TORCH_IMPL_FUNC() function
51 TORCH_IMPL_FUNC(pow_Tensor_Scalar_out) (const Tensor& base, const Scalar& exp, const Tensor& out) { in TORCH_IMPL_FUNC() function
61 TORCH_IMPL_FUNC(pow_Scalar_out) (const Scalar& base, const Tensor& exp, const Tensor& out) { in TORCH_IMPL_FUNC() function
|
| H A D | TensorAdvancedIndexing.cpp | 624 TORCH_IMPL_FUNC(index_out) in TORCH_IMPL_FUNC() function
871 TORCH_IMPL_FUNC(index_copy_out) in TORCH_IMPL_FUNC() function
942 TORCH_IMPL_FUNC(index_add_cpu_out) in TORCH_IMPL_FUNC() function
1214 TORCH_IMPL_FUNC(index_reduce_cpu_out) in TORCH_IMPL_FUNC() function
1638 TORCH_IMPL_FUNC(gather_out) in TORCH_IMPL_FUNC() function
1840 TORCH_IMPL_FUNC(scatter_src_out) in TORCH_IMPL_FUNC() function
1851 TORCH_IMPL_FUNC(scatter_value_out) in TORCH_IMPL_FUNC() function
1862 TORCH_IMPL_FUNC(scatter_reduce_out) in TORCH_IMPL_FUNC() function
1875 TORCH_IMPL_FUNC(scatter_value_reduce_out) in TORCH_IMPL_FUNC() function
1888 TORCH_IMPL_FUNC(scatter_add) in TORCH_IMPL_FUNC() function
[all …]
|
| H A D | SoftMax.cpp | 335 TORCH_IMPL_FUNC(softmax_cpu_out) in TORCH_IMPL_FUNC() function
363 TORCH_IMPL_FUNC(log_softmax_cpu_out) in TORCH_IMPL_FUNC() function
390 TORCH_IMPL_FUNC(softmax_backward_cpu_out) in TORCH_IMPL_FUNC() function
419 TORCH_IMPL_FUNC(log_softmax_backward_cpu_out) ( in TORCH_IMPL_FUNC() function
|
| H A D | Lerp.cpp | 39 TORCH_IMPL_FUNC(lerp_Tensor)( in TORCH_IMPL_FUNC() function
44 TORCH_IMPL_FUNC(lerp_Scalar)( in TORCH_IMPL_FUNC() function
|
| H A D | PointwiseOps.cpp | 49 TORCH_IMPL_FUNC(addcmul_out) in TORCH_IMPL_FUNC() function
58 TORCH_IMPL_FUNC(addcdiv_out) in TORCH_IMPL_FUNC() function
|
| H A D | AdaptiveMaxPooling2d.cpp | 73 TORCH_IMPL_FUNC(adaptive_max_pool2d_out_cpu) in TORCH_IMPL_FUNC() function
78 TORCH_IMPL_FUNC(adaptive_max_pool2d_backward_out_cpu) in TORCH_IMPL_FUNC() function
|
| H A D | UpSampleLinear1d.cpp | 64 TORCH_IMPL_FUNC(upsample_linear1d_out_cpu) ( in TORCH_IMPL_FUNC() function
74 TORCH_IMPL_FUNC(upsample_linear1d_backward_out_cpu) ( in TORCH_IMPL_FUNC() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/operations/ |
| H A D | BitwiseOps.mm | 343 TORCH_IMPL_FUNC(bitwise_and_out_mps)(const Tensor& self, const Tensor& other, const Tensor& output)… function
347 TORCH_IMPL_FUNC(bitwise_or_out_mps)(const Tensor& self, const Tensor& other, const Tensor& output) { function
351 TORCH_IMPL_FUNC(bitwise_xor_out_mps)(const Tensor& self, const Tensor& other, const Tensor& output)… function
355 TORCH_IMPL_FUNC(bitwise_not_out_mps)(const Tensor& self, const Tensor& output) { function
|
| H A D | Gamma.mm | 400 TORCH_IMPL_FUNC(lgamma_out_mps)(const Tensor& self, const Tensor& output_) { function
440 TORCH_IMPL_FUNC(digamma_out_mps)(const Tensor& self, const Tensor& output_) { function
479 TORCH_IMPL_FUNC(polygamma_out_mps)(const int64_t order, const Tensor& self, const Tensor& output_) { function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | UpSampleNearest1d.cu | 200 TORCH_IMPL_FUNC(upsample_nearest1d_out_cuda) ( in TORCH_IMPL_FUNC() function
210 TORCH_IMPL_FUNC(_upsample_nearest_exact1d_out_cuda) ( in TORCH_IMPL_FUNC() function
219 TORCH_IMPL_FUNC(upsample_nearest1d_backward_out_cuda) ( in TORCH_IMPL_FUNC() function
230 TORCH_IMPL_FUNC(_upsample_nearest_exact1d_backward_out_cuda) ( in TORCH_IMPL_FUNC() function
|
| H A D | Blas.cpp | 595 TORCH_IMPL_FUNC(addmm_out_cuda)(const Tensor& self, const Tensor& mat1, const Tensor& mat2, const S… in TORCH_IMPL_FUNC() function
599 TORCH_IMPL_FUNC(addmm_activation_out_cuda)(const Tensor& self, const Tensor& mat1, const Tensor& ma… in TORCH_IMPL_FUNC() function
603 TORCH_IMPL_FUNC(mm_out_cuda)(const Tensor& self, const Tensor& mat2, const Tensor& result) { in TORCH_IMPL_FUNC() function
607 TORCH_IMPL_FUNC(baddbmm_out_cuda)(const Tensor& self, const Tensor& batch1, const Tensor& batch2, c… in TORCH_IMPL_FUNC() function
614 TORCH_IMPL_FUNC(bmm_out_cuda)(const Tensor& batch1, const Tensor& batch2, const Tensor &result) { in TORCH_IMPL_FUNC() function
757 TORCH_IMPL_FUNC(addmv_out_cuda)(const Tensor &self, const Tensor &mat, const Tensor &vec, const Sca… in TORCH_IMPL_FUNC() function
|
| H A D | UpSampleNearest3d.cu | 295 TORCH_IMPL_FUNC(upsample_nearest3d_out_cuda) ( in TORCH_IMPL_FUNC() function
306 TORCH_IMPL_FUNC(_upsample_nearest_exact3d_out_cuda) ( in TORCH_IMPL_FUNC() function
316 TORCH_IMPL_FUNC(upsample_nearest3d_backward_out_cuda) ( in TORCH_IMPL_FUNC() function
328 TORCH_IMPL_FUNC(_upsample_nearest_exact3d_backward_out_cuda) ( in TORCH_IMPL_FUNC() function
|
| H A D | UpSampleNearest2d.cu | 446 TORCH_IMPL_FUNC(upsample_nearest2d_out_cuda) ( in TORCH_IMPL_FUNC() function
456 TORCH_IMPL_FUNC(_upsample_nearest_exact2d_out_cuda) ( in TORCH_IMPL_FUNC() function
466 TORCH_IMPL_FUNC(upsample_nearest2d_backward_out_cuda) ( in TORCH_IMPL_FUNC() function
477 TORCH_IMPL_FUNC(_upsample_nearest_exact2d_backward_out_cuda) ( in TORCH_IMPL_FUNC() function
|
| H A D | UpSampleBilinear2d.cu | 847 TORCH_IMPL_FUNC(upsample_bilinear2d_out_cuda) ( in TORCH_IMPL_FUNC() function
857 TORCH_IMPL_FUNC(upsample_bilinear2d_backward_out_cuda) ( in TORCH_IMPL_FUNC() function
872 TORCH_IMPL_FUNC(_upsample_bilinear2d_aa_out_cuda) ( in TORCH_IMPL_FUNC() function
884 TORCH_IMPL_FUNC(_upsample_bilinear2d_aa_backward_out_cuda) ( in TORCH_IMPL_FUNC() function
901 TORCH_IMPL_FUNC(_upsample_bicubic2d_aa_out_cuda) ( in TORCH_IMPL_FUNC() function
912 TORCH_IMPL_FUNC(_upsample_bicubic2d_aa_backward_out_cuda) ( in TORCH_IMPL_FUNC() function
|