Searched defs:tensors2 (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ForeachBinaryOpList.cu | 30 TensorList tensors2, in foreach_tensor_list_op()
60 TensorList tensors2, in foreach_tensor_list_op_()
82 TensorList tensors2, in all_types_complex_bool_half_bfloat16()
98 TensorList tensors2, in all_types_complex_bool_half_bfloat16_()
114 TensorList tensors2, in all_types_half_bfloat16()
129 TensorList tensors2, in all_types_complex_half_bfloat16_()
144 TensorList tensors2, in all_types_half_bfloat16_()
159 TensorList tensors2, in all_types_complex_half_bfloat16()
|
| H A D | ForeachTernaryOp.cu | 27 TensorList tensors2, in foreach_tensor_lerp_ternary_cuda()
64 TensorList tensors2, in foreach_tensor_lerp_ternary_cuda_()
94 TensorList tensors2, in foreach_tensor_lerp_list_cuda()
132 TensorList tensors2, in foreach_tensor_lerp_list_cuda_()
|
| H A D | ForeachPointwiseOp.cu | 28 TensorList tensors2, in foreach_pointwise_op()
67 TensorList tensors2, in foreach_pointwise_op_()
98 TensorList tensors2, in foreach_pointwise_op_()
130 TensorList tensors2, in foreach_pointwise_op()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ForeachUtils.h | 55 TensorList tensors2) { in check_foreach_api_restrictions()
68 TensorList tensors2, in check_foreach_api_restrictions()
89 TensorList tensors2, in check_foreach_api_restrictions()
|