Add restrictions on multi-input convolutions (#521)

kevinch-nv · web-flow · commit fbebb144744b · 2020-09-02T14:59:20.000-07:00
diff --git a/builtin_op_importers.cpp b/builtin_op_importers.cpp
@@ -430,7 +430,7 @@ DEFINE_BUILTIN_OP_IMPORTER(Conv)
     ASSERT(inputs.at(0).is_tensor(), ErrorCode::kUNSUPPORTED_NODE);
     if (inputs.at(1).is_tensor())
     {
-        ASSERT(inputs.at(1).is_tensor(), ErrorCode::kUNSUPPORTED_NODE);
+        ASSERT(ctx->network()->hasExplicitPrecision() && "TensorRT only supports multi-input conv for explicit precision QAT networks!", ErrorCode::kUNSUPPORTED_NODE);
         if (inputs.size() == 3)
         {
             ASSERT(inputs.at(2).is_weights(), ErrorCode::kUNSUPPORTED_NODE);

Original file line number	Diff line number	Diff line change
`@@ -430,7 +430,7 @@ DEFINE_BUILTIN_OP_IMPORTER(Conv)`
`430`	`430`	`ASSERT(inputs.at(0).is_tensor(), ErrorCode::kUNSUPPORTED_NODE);`
`431`	`431`	`if (inputs.at(1).is_tensor())`
`432`	`432`	`{`
`433`		`- ASSERT(inputs.at(1).is_tensor(), ErrorCode::kUNSUPPORTED_NODE);`
	`433`	`+ ASSERT(ctx->network()->hasExplicitPrecision() && "TensorRT only supports multi-input conv for explicit precision QAT networks!", ErrorCode::kUNSUPPORTED_NODE);`
`434`	`434`	`if (inputs.size() == 3)`
`435`	`435`	`{`
`436`	`436`	`ASSERT(inputs.at(2).is_weights(), ErrorCode::kUNSUPPORTED_NODE);`