From 610ce3aea2751e010c1cba1f80e94215fc1f0b79 Mon Sep 17 00:00:00 2001 From: Superjomn Date: Mon, 22 Apr 2019 11:29:10 +0800 Subject: [PATCH] add io_copy_op.h --- paddle/fluid/lite/operators/io_copy_op.h | 40 ++++++++++++++++++++++++ 1 file changed, 40 insertions(+) create mode 100644 paddle/fluid/lite/operators/io_copy_op.h diff --git a/paddle/fluid/lite/operators/io_copy_op.h b/paddle/fluid/lite/operators/io_copy_op.h new file mode 100644 index 00000000000..0df41501f76 --- /dev/null +++ b/paddle/fluid/lite/operators/io_copy_op.h @@ -0,0 +1,40 @@ +// 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 "paddle/fluid/lite/core/op_lite.h" + +namespace paddle { +namespace lite { +namespace operators { + +class IoCopyOp : public OpLite { + public: + explicit IoCopyOp(const std::string &type) : OpLite(type) {} + bool CheckShape() const override; + bool InferShape() const override; + bool Run() override; + std::string DebugString() const override; + + protected: + bool AttachImpl(const framework::OpDesc &opdesc, lite::Scope *scope) override; + + private: + operators::IoCopyParam param_; +}; + +} // namespace operators +} // namespace lite +} // namespace paddle -- GitLab