diff --git a/paddle/fluid/operators/ngraph/ops/lrn_op.h b/paddle/fluid/operators/ngraph/ops/lrn_op.h new file mode 100644 index 0000000000000000000000000000000000000000..68a0eea08928ae6bfdae50bef9a3f5c2fddde9c8 --- /dev/null +++ b/paddle/fluid/operators/ngraph/ops/lrn_op.h @@ -0,0 +1,54 @@ +// Copyright (c) 2019 PaddlePaddle Authors. All Rights Reserved. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +#pragma once + +#include +#include +#include +#include + +#include "ngraph/ngraph.hpp" +#include "paddle/fluid/operators/ngraph/ops/op_bridge.h" +#include "paddle/fluid/platform/ngraph_helper.h" + +namespace paddle { +namespace operators { +namespace ngraphs { +static void BuildLrnNode( + const std::shared_ptr& op, + std::shared_ptr< + std::unordered_map>> + ngb_node_map) { + auto input = platform::GetInputNode(op, "X", ngb_node_map); + + auto op_attrs = framework::AttrReader(op->Attrs()); + const int n = op_attrs.Get("n"); + const float alpha = op_attrs.Get("alpha") * static_cast(n); + const float beta = op_attrs.Get("beta"); + const float k = op_attrs.Get("k"); + + auto lrn_out = std::make_shared(input, alpha, beta, k, n); + std::shared_ptr mid_out = paddle::platform::CreateConstant( + input->get_element_type(), input->get_shape(), {k}); + + platform::SetOutputNode(op, "MidOut", mid_out, ngb_node_map); + platform::SetOutputNode(op, "Out", lrn_out, ngb_node_map); +} + +} // namespace ngraphs +} // namespace operators +} // namespace paddle + +REGISTER_NG_OP(lrn, BuildLrnNode); diff --git a/paddle/fluid/platform/ngraph_helper.h b/paddle/fluid/platform/ngraph_helper.h index e74f57a79a66ea8fe8c9b972a9a2ec9d722731eb..9e6521653b80abec1c5212f5deb84153335c2a9c 100644 --- a/paddle/fluid/platform/ngraph_helper.h +++ b/paddle/fluid/platform/ngraph_helper.h @@ -16,7 +16,9 @@ limitations under the License. */ #pragma once #include +#include #include +#include #include #include "ngraph/ngraph.hpp" @@ -103,6 +105,25 @@ std::shared_ptr GetOutputNode( return GetNode(op, name, op->Outputs(), ngb_node_map); } +template +std::shared_ptr CreateConstant(const ngraph::element::Type& type, + ngraph::Shape shape, + std::initializer_list values) { + std::shared_ptr result; + if (values.size() == 1 && shape != ngraph::Shape{} && // NOLINT + shape != ngraph::Shape{1}) { + result = std::make_shared(type, ngraph::Shape{}, + std::vector{values}); + ngraph::AxisSet axis_set; + for (size_t i = 0; i < shape.size(); ++i) axis_set.insert(i); + result = std::make_shared(result, shape, axis_set); + } else { + result = std::make_shared(type, shape, + std::vector{values}); + } + return result; +} + void SetOutputNode( const std::shared_ptr& op, const std::string name, std::shared_ptr node, diff --git a/python/paddle/fluid/tests/unittests/ngraph/test_lrn_ngraph_op.py b/python/paddle/fluid/tests/unittests/ngraph/test_lrn_ngraph_op.py new file mode 100644 index 0000000000000000000000000000000000000000..4c998c6ca2ec3bba8d7c3257ed3fe4fddc70a46a --- /dev/null +++ b/python/paddle/fluid/tests/unittests/ngraph/test_lrn_ngraph_op.py @@ -0,0 +1,29 @@ +# Copyright (c) 2019 PaddlePaddle Authors. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + +from __future__ import print_function + +import unittest +from paddle.fluid.tests.unittests.test_lrn_op import TestLRNOp + + +class TestLRNNGRAPHOp(TestLRNOp): + def test_check_output(self): + self.check_output(atol=0.002) + + +del TestLRNOp + +if __name__ == '__main__': + unittest.main()