Handle node with no defined input

Signed-off-by: Serge Panev <[email protected]>
apache · Jun 29, 2020 · 6d4dae1 · 6d4dae1
1 parent 6944428
commit 6d4dae1
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
@@ -320,7 +320,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;