[moco/tf] Add graph validation (#3653)
author박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Mon, 3 Jun 2019 03:10:53 +0000 (12:10 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Mon, 3 Jun 2019 03:10:53 +0000 (12:10 +0900)
* [moco/tf] Add graph validation

This will add graph validation after converting the graph from TensorFlow to loco.

Signed-off-by: SaeHie Park <saehie.park@samsung.com>
* remove nullptr

* add assert

contrib/moco/lib/frontend/tf/src/Frontend.cpp

index 05930b5..4a32c26 100644 (file)
@@ -20,6 +20,7 @@
 #include "GraphBuilderContext.h"
 #include "GraphBuilderRegistry.h"
 
+#include <loco/IR/Verifier.h>
 #include <cwrap/Fildes.h>
 #include <stdex/Memory.h>
 
@@ -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