Fix a bug in MXNet-TensorRT (apache#18870)

Signed-off-by: Serge Panev <[email protected]>
zheyuye · Aug 7, 2020 · 708a900 · 708a900
1 parent d101c3c
commit 708a900
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/operator/subgraph/tensorrt/tensorrt-inl.h b/src/operator/subgraph/tensorrt/tensorrt-inl.h
@@ -302,7 +302,7 @@ class TensorrtProperty : public SubgraphProperty {
     TRTParam param;
     std::ostringstream params_oss;
     for (auto &param_name : new_sym.ListInputNames(nnvm::Symbol::kAll)) {
-      NDArray *cache;
+      NDArray *cache = nullptr;
       auto it_args = in_args_dict.find(param_name);
       if (it_args != in_args_dict.end()) {
         cache = it_args->second;