diff --git a/viewer/dot_graph.cpp b/viewer/dot_graph.cpp
index e281c195c186a281421e822d30fcf167a6ad63d2..ac2d868962fed75034123d3e1ebed68d6f6978a6 100644 (file)
--- a/viewer/dot_graph.cpp
+++ b/viewer/dot_graph.cpp
/******************************************************************************
- * Copyright (c) 2017-18, Texas Instruments Incorporated - http://www.ti.com/
+ * Copyright (c) 2017-2018, Texas Instruments Incorporated - http://www.ti.com/
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
else if (p.poolingType == TIDL_AveragePooling)
s += " Average";
- s+= "\\n" + std::to_string(p.kernelW) + "x" + std::to_string(p.kernelH);
+ if (p.kernelW != 0 && p.kernelH != 0)
+ s+= "\\n" + std::to_string(p.kernelW) + "x" + std::to_string(p.kernelH);
return s;
}