Handle node with no defined input

Signed-off-by: Serge Panev <spanev@nvidia.com>
apache · Jun 10, 2020 · dd11df7 · dd11df7
1 parent 8d0c8f0
commit dd11df7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/operator/subgraph/tensorrt/tensorrt-inl.h b/src/operator/subgraph/tensorrt/tensorrt-inl.h
@@ -321,7 +321,9 @@ class TensorrtProperty : public SubgraphProperty {
       }
     }
     auto tensorrt_params_names = params_oss.str();
-    tensorrt_params_names.pop_back();
+    if (!tensorrt_params_names.empty()) {
+      tensorrt_params_names.pop_back();
+    }
     n->attrs.parsed = param;
     n->attrs.dict["subgraph_params_names"] = tensorrt_params_names;
     return n;