Home
last modified time | relevance | path

Searched defs:TORCH_IMPL_FUNC (Results 1 – 25 of 68) sorted by relevance

123

/external/pytorch/aten/src/ATen/native/mps/operations/
DBitwiseOps.mm343 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
DGamma.mm400 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
/external/pytorch/aten/src/ATen/native/
DUpSampleNearest1d.cpp80 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
DActivation.cpp270 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 …]
DUpSampleNearest2d.cpp104 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
DUpSampleBilinear2d.cpp108 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
DUpSampleNearest3d.cpp113 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
DLerp.cpp39 TORCH_IMPL_FUNC(lerp_Tensor)( in TORCH_IMPL_FUNC() function
44 TORCH_IMPL_FUNC(lerp_Scalar)( in TORCH_IMPL_FUNC() function
DPointwiseOps.cpp49 TORCH_IMPL_FUNC(addcmul_out) in TORCH_IMPL_FUNC() function
58 TORCH_IMPL_FUNC(addcdiv_out) in TORCH_IMPL_FUNC() function
DTensorCompare.cpp646 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 …]
DAdaptiveMaxPooling2d.cpp73 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
DUpSampleLinear1d.cpp64 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
DUpSampleTrilinear3d.cpp70 TORCH_IMPL_FUNC(upsample_trilinear3d_out_cpu) ( in TORCH_IMPL_FUNC() function
82 TORCH_IMPL_FUNC(upsample_trilinear3d_backward_out_cpu) ( in TORCH_IMPL_FUNC() function
DPow.cpp47 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
DUpSampleBicubic2d.cpp226 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
DReplicationPadding.cpp271 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
DReflectionPad.cpp312 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
DAveragePool2d.cpp152 TORCH_IMPL_FUNC(avg_pool2d_out_cpu) in TORCH_IMPL_FUNC() function
178 TORCH_IMPL_FUNC(avg_pool2d_backward_out_cpu) ( in TORCH_IMPL_FUNC() function
DBinaryOps.cpp433 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 …]
DDilatedMaxPool2d.cpp162 TORCH_IMPL_FUNC(max_pool2d_with_indices_out_cpu) in TORCH_IMPL_FUNC() function
194 TORCH_IMPL_FUNC(max_pool2d_with_indices_backward_out_cpu) in TORCH_IMPL_FUNC() function
/external/pytorch/aten/src/ATen/native/cuda/
DUpSampleNearest1d.cu200 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
DActivation.cpp96 TORCH_IMPL_FUNC(gelu_out_cuda) ( in TORCH_IMPL_FUNC() function
102 TORCH_IMPL_FUNC(gelu_backward_out_cuda) ( in TORCH_IMPL_FUNC() function
DUpSampleNearest3d.cu295 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
DTriangularOps.cu147 TORCH_IMPL_FUNC(tril_cuda)(const Tensor& self, int64_t k, const Tensor &result) { in TORCH_IMPL_FUNC() function
153 TORCH_IMPL_FUNC(triu_cuda)(const Tensor& self, int64_t k, const Tensor &result) { in TORCH_IMPL_FUNC() function
DBlas.cpp595 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

123