Add StopGradient to the graph serializer (#1410)

e362646e · Jaikrishnan Menon · Robert Kimball · c449005a · e362646e
Commit e362646e authored Aug 14, 2018 by Jaikrishnan Menon Committed by Robert Kimball Aug 14, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

serializer.cpp src/ngraph/serializer.cpp +5 -0

No files found.
--- a/src/ngraph/serializer.cpp
+++ b/src/ngraph/serializer.cpp
@@ -83,6 +83,7 @@
 #include "ngraph/op/slice.hpp"
 #include "ngraph/op/softmax.hpp"
 #include "ngraph/op/sqrt.hpp"
+#include "ngraph/op/stop_gradient.hpp"
 #include "ngraph/op/subtract.hpp"
 #include "ngraph/op/sum.hpp"
 #include "ngraph/op/tan.hpp"
@@ -900,6 +901,10 @@ static shared_ptr<ngraph::Function>
            {
                node = make_shared<op::Tanh>(args[0]);
            }
+            else if (node_op == "StopGradient")
+            {
+                node = make_shared<op::StopGradient>(args[0]);
+            }
            else
            {
                stringstream ss;