Apply suggestions from code review

Co-authored-by: Nicolas Hug <[email protected]>

Author: AntoineSimoulin

torchvision/transforms/v2/functional/_geometry.py

@@ -452,7 +452,7 @@ def _parallelogram_to_bounding_boxes(parallelogram: torch.Tensor) -> torch.Tenso
                      The output maintains the same dtype as the input.
     """
     dtype = parallelogram.dtype
-    acceptable_dtypes = [torch.float32]
+    acceptable_dtypes = [torch.float32, torch.float64]
     need_cast = dtype not in acceptable_dtypes
     if need_cast:
         # Up-case to avoid overflow for square operations

torchvision/transforms/v2/functional/_meta.py

@@ -251,7 +251,7 @@ def _xyxyxyxy_to_xywhr(xyxyxyxy: torch.Tensor, inplace: bool) -> torch.Tensor:
         xyxyxyxy = xyxyxyxy.clone()
 
     dtype = xyxyxyxy.dtype
-    acceptable_dtypes = [torch.float32]  # Ensure consistency between CPU and GPU.
+    acceptable_dtypes = [torch.float32, torch.float64]  # Ensure consistency between CPU and GPU.
     need_cast = dtype not in acceptable_dtypes
     if need_cast:
         # Up-case to avoid overflow for square operations