From 93b948ffa8976baa5dd1bd076dadd9fac9d8955e Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princip?= =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Mon, 3 Jun 2019 12:10:53 +0900 Subject: [PATCH] [moco/tf] Add graph validation (#3653) * [moco/tf] Add graph validation This will add graph validation after converting the graph from TensorFlow to loco. Signed-off-by: SaeHie Park * remove nullptr * add assert --- contrib/moco/lib/frontend/tf/src/Frontend.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/contrib/moco/lib/frontend/tf/src/Frontend.cpp b/contrib/moco/lib/frontend/tf/src/Frontend.cpp index 05930b5..4a32c26 100644 --- a/contrib/moco/lib/frontend/tf/src/Frontend.cpp +++ b/contrib/moco/lib/frontend/tf/src/Frontend.cpp @@ -20,6 +20,7 @@ #include "GraphBuilderContext.h" #include "GraphBuilderRegistry.h" +#include #include #include @@ -215,6 +216,9 @@ void convert_graph(const moco::tf::ModelSignature &signature, tensorflow::GraphD graph_output->name(output); graph_output->node(push_node); } + + // validate graph + assert(loco::valid(graph)); } } // namespace -- 2.7.4