[TFTRT] Add Dynamic Shape Testing for ConvertConv3D

tensorflow · Feb 23, 2021 · 20f5356 · 20f5356
1 parent 7ca8021
commit 20f5356
Show file tree

Hide file tree

Showing 2 changed files with 237 additions and 169 deletions.
diff --git a/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc b/tensorflow/compiler/tf2tensorrt/convert/convert_nodes.cc
@@ -3318,6 +3318,13 @@ Status ConvertConv3DHelper(OpConverterParams* params, int group,
     }
   }
 
+  // Channel dim must be static for Conv3D since we use that value for
+  // num_groups at build time.
+  if (tensor->getDimensions().d[c_index] == -1) {
+    return errors::InvalidArgument("Channel dimension must be static, at ",
+                                   node_def.name());
+  }
+
   // Finished validation checks
   if (params->validation_only) return Status::OK();