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 0000000000000000000000000000000000000000..0df41501f760617e0a995bc0fbaea0e352b7f2b6 --- /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