From 1ce7b45b9e0a6cec0331483c07429cf57872383b Mon Sep 17 00:00:00 2001 From: baojun <32073718+baojun-nervana@users.noreply.github.com> Date: Tue, 14 May 2019 21:55:57 -0700 Subject: [PATCH] NGraph Added fill_zeros_like op test=develop (#17295) --- .../operators/ngraph/ops/fill_zeros_like_op.h | 45 +++++++++++++++++++ .../ngraph/test_fill_zeros_like_ngraph_op.py | 20 +++++++++ 2 files changed, 65 insertions(+) create mode 100644 paddle/fluid/operators/ngraph/ops/fill_zeros_like_op.h create mode 100644 python/paddle/fluid/tests/unittests/ngraph/test_fill_zeros_like_ngraph_op.py diff --git a/paddle/fluid/operators/ngraph/ops/fill_zeros_like_op.h b/paddle/fluid/operators/ngraph/ops/fill_zeros_like_op.h new file mode 100644 index 00000000000..163734be877 --- /dev/null +++ b/paddle/fluid/operators/ngraph/ops/fill_zeros_like_op.h @@ -0,0 +1,45 @@ +/*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 BuildFillZerosLikeNode( + const std::shared_ptr& op, + std::shared_ptr< + std::unordered_map>> + ngb_node_map) { + auto x = platform::GetInputNode(op, "X", ngb_node_map); + auto out = paddle::platform::CreateConstant(x->get_element_type(), + x->get_shape(), {0}); + platform::SetOutputNode(op, "Out", out, ngb_node_map); +} + +} // namespace ngraphs +} // namespace operators +} // namespace paddle + +REGISTER_NG_OP(fill_zeros_like, BuildFillZerosLikeNode); diff --git a/python/paddle/fluid/tests/unittests/ngraph/test_fill_zeros_like_ngraph_op.py b/python/paddle/fluid/tests/unittests/ngraph/test_fill_zeros_like_ngraph_op.py new file mode 100644 index 00000000000..45e6af7f961 --- /dev/null +++ b/python/paddle/fluid/tests/unittests/ngraph/test_fill_zeros_like_ngraph_op.py @@ -0,0 +1,20 @@ +# 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_fill_zeros_like_op import TestFillZerosLikeOp + +if __name__ == "__main__": + unittest.main() -- GitLab