From 32900243167b1cc3bdf39e6a1ce1ab44ccca70b7 Mon Sep 17 00:00:00 2001 From: Superjomn Date: Mon, 22 Apr 2019 11:02:09 +0800 Subject: [PATCH] add io_copy_op.cc --- paddle/fluid/lite/operators/io_copy_op.cc | 43 +++++++++++++++++++++++ 1 file changed, 43 insertions(+) create mode 100644 paddle/fluid/lite/operators/io_copy_op.cc diff --git a/paddle/fluid/lite/operators/io_copy_op.cc b/paddle/fluid/lite/operators/io_copy_op.cc new file mode 100644 index 0000000000..0169d85ba1 --- /dev/null +++ b/paddle/fluid/lite/operators/io_copy_op.cc @@ -0,0 +1,43 @@ +// 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. + +#include "paddle/fluid/lite/operators/io_copy_op.h" +#include "paddle/fluid/lite/core/op_registry.h" + +namespace paddle { +namespace lite { +namespace operators { + +bool IoCopyOp::CheckShape() const { + CHECK_OR_FALSE(param_.x); + CHECK_OR_FALSE(param_.y); + return true; +} +bool IoCopyOp::InferShape() const { return true; } +bool IoCopyOp::Run() { return OpLite::Run(); } +bool IoCopyOp::AttachImpl(const paddle::framework::OpDesc &opdesc, + paddle::lite::Scope *scope) { + auto x = opdesc.Input("Input").front(); + auto out = opdesc.Output("Out").front(); + param_.x = GetTensor(scope, x); + param_.y = GetMutableTensor(scope, out); + return true; +} +std::string IoCopyOp::DebugString() const { return "io_copy_op"; } + +} // namespace operators +} // namespace lite +} // namespace paddle + +REGISTER_LITE_OP(io_copy, paddle::lite::operators::IoCopyOp); -- GitLab