Added TensorRT as a target and fixed Reshape op conversion for TensorRT (#1388)

Signed-off-by: Tom Wildenhain <[email protected]>

Co-authored-by: Guenther Schmuelling <[email protected]>

Author: TomWildenhain-Microsoft

tf2onnx/constants.py

@@ -28,7 +28,8 @@
 TARGET_RS5 = "rs5"
 TARGET_RS6 = "rs6"
 TARGET_CAFFE2 = "caffe2"
-POSSIBLE_TARGETS = [TARGET_RS4, TARGET_RS5, TARGET_RS6, TARGET_CAFFE2]
+TARGET_TENSORRT = "tensorrt"
+POSSIBLE_TARGETS = [TARGET_RS4, TARGET_RS5, TARGET_RS6, TARGET_CAFFE2, TARGET_TENSORRT]
 DEFAULT_TARGET = []
 
 NCHW_TO_NHWC = [0, 2, 3, 1]

tf2onnx/onnx_opset/nn.py

@@ -1062,7 +1062,8 @@ def version_11(cls, ctx, node, **kwargs):
         if "align_corners" in node.attr and node.attr["align_corners"].i:
             transformation_mode = "align_corners"
         if "half_pixel_centers" in node.attr and node.attr["half_pixel_centers"].i:
-            if node.type == "ResizeNearestNeighbor":
+            if node.type == "ResizeNearestNeighbor" and not ctx.is_target(constants.TARGET_TENSORRT):
+                # TensorRT only supports nearest_mode = "floor" for mode = "nearest"
                 transformation_mode = "half_pixel"
                 nearest_mode = "round_prefer_ceil"
             else: