Searched defs:tensors1 (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | ForeachBinaryOpList.cu | 29 TensorList tensors1, in foreach_tensor_list_op()
59 TensorList tensors1, in foreach_tensor_list_op_()
81 TensorList tensors1, in all_types_complex_bool_half_bfloat16()
97 TensorList tensors1, in all_types_complex_bool_half_bfloat16_()
113 TensorList tensors1, in all_types_half_bfloat16()
128 TensorList tensors1, in all_types_complex_half_bfloat16_()
143 TensorList tensors1, in all_types_half_bfloat16_()
158 TensorList tensors1, in all_types_complex_half_bfloat16()
|
| H A D | ForeachTernaryOp.cu | 26 TensorList tensors1, in foreach_tensor_lerp_ternary_cuda()
63 TensorList tensors1, in foreach_tensor_lerp_ternary_cuda_()
93 TensorList tensors1, in foreach_tensor_lerp_list_cuda()
131 TensorList tensors1, in foreach_tensor_lerp_list_cuda_()
|
| H A D | ForeachPointwiseOp.cu | 27 TensorList tensors1, in foreach_pointwise_op()
66 TensorList tensors1, in foreach_pointwise_op_()
97 TensorList tensors1, in foreach_pointwise_op_()
129 TensorList tensors1, in foreach_pointwise_op()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ForeachUtils.h | 54 TensorList tensors1, in check_foreach_api_restrictions()
67 TensorList tensors1, in check_foreach_api_restrictions()
88 TensorList tensors1, in check_foreach_api_restrictions()
|