chore: import upstream snapshot with attribution

python/paddle/tensorrt/impls/vision.py

@@ -0,0 +1,48 @@
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import tensorrt as trt
+
+from paddle.tensorrt.converter_utils import set_layer_name
+from paddle.tensorrt.register import converter_registry
+
+
+@converter_registry.register("pd_op.grid_sample")
+def grid_sample_converter(network, paddle_op, inputs):
+    input_tensor, grid_tensor = inputs
+    padding = paddle_op.attrs().get("paddings", [0, 0])
+
+    mode = paddle_op.attrs().get("mode", "bilinear")
+    padding_mode = paddle_op.attrs().get("padding_mode", "zeros")
+    align_corners = paddle_op.attrs().get("align_corners", True)
+
+    if padding_mode == "zeros":
+        sample_mode = trt.SampleMode.FILL
+    elif padding_mode == "border":
+        sample_mode = trt.SampleMode.CLAMP
+    elif padding_mode == "reflection":
+        sample_mode = trt.SampleMode.REFLECT
+
+    if mode == "nearest":
+        interpolation_mode = trt.InterpolationMode.NEAREST
+    elif mode == "bilinear":
+        interpolation_mode = trt.InterpolationMode.LINEAR
+
+    grid_sample_layer = network.add_grid_sample(input_tensor, grid_tensor)
+
+    grid_sample_layer.interpolation_mode = interpolation_mode
+    grid_sample_layer.align_corners = align_corners
+    grid_sample_layer.sample_mode = sample_mode
+    set_layer_name(grid_sample_layer, paddle_op)
+    return grid_sample_layer.get_output(0)