refactor(//core/conversion): Run output checking on both converters and

narendasan · narendasan · commit 9d0bdafff236 · 2020-05-03T20:58:54.000-07:00
evaluators

Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
Signed-off-by: Naren Dasan &lt;narens@nvidia.com&gt;
diff --git a/core/conversion/conversion.cpp b/core/conversion/conversion.cpp
@@ -214,9 +214,7 @@ void ConvertBlockToNetDef(ConversionCtx* ctx, const torch::jit::Block* b, Conver
     }
 
     for (const auto n : nodes) {
-        if (converters::node_is_convertable(n)) {
-            ctx->CheckLayerAddition(n);
-        }
+        ctx->CheckLayerAddition(n);
     }
 
     auto outputs = b->outputs();

Original file line number	Diff line number	Diff line change
`@@ -214,9 +214,7 @@ void ConvertBlockToNetDef(ConversionCtx* ctx, const torch::jit::Block* b, Conver`
`214`	`214`	`}`
`215`	`215`
`216`	`216`	`for (const auto n : nodes) {`
`217`		`- if (converters::node_is_convertable(n)) {`
`218`		`- ctx->CheckLayerAddition(n);`
`219`		`- }`
	`217`	`+ ctx->CheckLayerAddition(n);`
`220`	`218`	`}`
`221`	`219`
`222`	`220`	`auto outputs = b->outputs();`