From b29acec815f8ebb1f2ca33d28b010751f4b132e5 Mon Sep 17 00:00:00 2001 From: mozga-intel Date: Wed, 27 Feb 2019 07:39:03 +0100 Subject: [PATCH] Register sum operator (#15889) test=develop --- paddle/fluid/operators/ngraph/ops/sum_op.h | 3 +++ 1 file changed, 3 insertions(+) diff --git a/paddle/fluid/operators/ngraph/ops/sum_op.h b/paddle/fluid/operators/ngraph/ops/sum_op.h index 97f4ce64aa5..ab8cdb8f4d8 100644 --- a/paddle/fluid/operators/ngraph/ops/sum_op.h +++ b/paddle/fluid/operators/ngraph/ops/sum_op.h @@ -18,6 +18,7 @@ limitations under the License. */ #include #include "ngraph/ngraph.hpp" +#include "paddle/fluid/operators/ngraph/ops/op_bridge.h" #include "paddle/fluid/platform/ngraph_helper.h" namespace paddle { @@ -53,3 +54,5 @@ void BuildSumNode( } // namespace ngraphs } // namespace operators } // namespace paddle + +REGISTER_NG_OP(sum, BuildSumNode); -- GitLab