Home
last modified time | relevance | path

Searched refs:check_foreach_api_restrictions (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DForeachOpsKernels.cpp77 check_foreach_api_restrictions(tensors); \
93 check_foreach_api_restrictions(tensors); \
114 check_foreach_api_restrictions(tensors); \
130 check_foreach_api_restrictions(tensors); \
144 check_foreach_api_restrictions(tensors); \
153 check_foreach_api_restrictions(tensors); \
167 check_foreach_api_restrictions(tensors, scalars); \
176 check_foreach_api_restrictions(tensors, scalars); \
189 check_foreach_api_restrictions(tensors1, tensors2); \
202 check_foreach_api_restrictions(tensors1, tensors2); \
[all …]
H A DForeachUtils.h40 inline void check_foreach_api_restrictions(TensorList tensors) { in check_foreach_api_restrictions() function
44 inline void check_foreach_api_restrictions( in check_foreach_api_restrictions() function
47 check_foreach_api_restrictions(tensors); in check_foreach_api_restrictions()
53 inline void check_foreach_api_restrictions( in check_foreach_api_restrictions() function
66 inline void check_foreach_api_restrictions( in check_foreach_api_restrictions() function
87 inline void check_foreach_api_restrictions( in check_foreach_api_restrictions() function
92 check_foreach_api_restrictions(tensors1, tensors2, tensors3); in check_foreach_api_restrictions()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DForeachBinaryOpScalar.cu144 check_foreach_api_restrictions(tensors); \
155 check_foreach_api_restrictions(tensors); \
183 check_foreach_api_restrictions(exponent); in foreach_scalar_pow_list_kernel_cuda()
213 check_foreach_api_restrictions(tensors); in foreach_tensor_div_scalar_kernel_cuda_()
226 check_foreach_api_restrictions(tensors); in foreach_tensor_div_scalar_kernel_cuda()
241 check_foreach_api_restrictions(tensors); in foreach_tensor_sub_scalar_kernel_cuda_()
260 check_foreach_api_restrictions(tensors); in foreach_tensor_sub_scalar_kernel_cuda()
H A DForeachUnaryOp.cu217 check_foreach_api_restrictions(tensors); \
225 check_foreach_api_restrictions(tensors); \
318 check_foreach_api_restrictions(tensors); in OP_CUSTOM_FUNCTOR()
332 check_foreach_api_restrictions(tensors); in foreach_tensor_neg_cuda_()
356 check_foreach_api_restrictions(tensors); in foreach_tensor_abs_cuda()
369 check_foreach_api_restrictions(tensors); in foreach_tensor_abs_cuda_()
382 check_foreach_api_restrictions(tensors); in foreach_tensor_zero_cuda_()
H A DForeachPointwiseOp.cu172 check_foreach_api_restrictions(input, tensors1, tensors2); \
188 check_foreach_api_restrictions(input, tensors1, tensors2); \
205 check_foreach_api_restrictions(input, tensors1, tensors2, scalars); \
221 check_foreach_api_restrictions(input, tensors1, tensors2, scalars); \
239 check_foreach_api_restrictions(input, tensors1, tensors2, scalars); \
255 check_foreach_api_restrictions(input, tensors1, tensors2, scalars); \
H A DForeachTernaryOp.cu29 check_foreach_api_restrictions(tensors1, tensors2, tensors3); in foreach_tensor_lerp_ternary_cuda()
66 check_foreach_api_restrictions(tensors1, tensors2, tensors3); in foreach_tensor_lerp_ternary_cuda_()
96 check_foreach_api_restrictions(tensors1, tensors2); in foreach_tensor_lerp_list_cuda()
134 check_foreach_api_restrictions(tensors1, tensors2); in foreach_tensor_lerp_list_cuda_()
H A DForeachBinaryOpScalarTensor.cu106 check_foreach_api_restrictions(tensors); \
123 check_foreach_api_restrictions(tensors); \
137 check_foreach_api_restrictions(tensors); \
149 check_foreach_api_restrictions(tensors); \
H A DForeachBinaryOpScalarList.cu147 check_foreach_api_restrictions(tensors, scalars); \
158 check_foreach_api_restrictions(tensors, scalars); \
195 check_foreach_api_restrictions(tensors, scalars); in foreach_tensor_sub_scalarlist_kernel_cuda_()
217 check_foreach_api_restrictions(tensors, scalars); in foreach_tensor_sub_scalarlist_kernel_cuda()
H A DForeachBinaryOpList.cu174 check_foreach_api_restrictions(tensors1, tensors2); \
185 check_foreach_api_restrictions(tensors1, tensors2); \
197 check_foreach_api_restrictions(tensors1, tensors2); \
208 check_foreach_api_restrictions(tensors1, tensors2); \
397 check_foreach_api_restrictions(self, src); in foreach_tensor_copy_list_kernel_cuda_()
H A DForeachReduceOp.cu133 check_foreach_api_restrictions(tensors); in foreach_tensor_max_cuda()
420 check_foreach_api_restrictions(tensors); in foreach_tensor_norm_cuda()
H A DAmpKernels.cu104 check_foreach_api_restrictions(scaled_grads); in _amp_foreach_non_finite_check_and_unscale_cuda_()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DAmpGradScalerKernels.cpp51 at::native::check_foreach_api_restrictions(scaled_grads); in _amp_foreach_non_finite_check_and_unscale_cpu_kernel()