[fbsync] fix elastic tests (#7841)

Reviewed By: matteobettini

Differential Revision: D48642293

fbshipit-source-id: 8e14e1e103ec518ff13e7d59e514120df9d74f1e

test/test_transforms_v2_refactored.py

@@ -2279,7 +2279,7 @@ def test_kernel_image_tensor(self, param, value, dtype, device):
         image = make_image_tensor(dtype=dtype, device=device)
 
         check_kernel(
-            F.elastic_image_tensor,
+            F.elastic_image,
             image,
             displacement=self._make_displacement(image),
             **{param: value},
@@ -2320,9 +2320,9 @@ def test_functional(self, make_input):
     @pytest.mark.parametrize(
         ("kernel", "input_type"),
         [
-            (F.elastic_image_tensor, torch.Tensor),
-            (F.elastic_image_pil, PIL.Image.Image),
-            (F.elastic_image_tensor, datapoints.Image),
+            (F.elastic_image, torch.Tensor),
+            (F._elastic_image_pil, PIL.Image.Image),
+            (F.elastic_image, datapoints.Image),
             (F.elastic_bounding_boxes, datapoints.BoundingBoxes),
             (F.elastic_mask, datapoints.Mask),
             (F.elastic_video, datapoints.Video),