From a88b1aa2be95e55075b6a08ce570a6ff638a99db Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princip?= =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Wed, 10 Jul 2019 07:07:06 +0900 Subject: [PATCH] [moco/tf] Introduce StrideData (#4158) This will introduce StrideData used as annotation for TFConv2D stride Signed-off-by: SaeHie Park --- contrib/moco-tf/src/Annotations/StrideData.h | 48 ++++++++++++++++++++++++++++ 1 file changed, 48 insertions(+) create mode 100644 contrib/moco-tf/src/Annotations/StrideData.h diff --git a/contrib/moco-tf/src/Annotations/StrideData.h b/contrib/moco-tf/src/Annotations/StrideData.h new file mode 100644 index 0000000..fb9a4b3 --- /dev/null +++ b/contrib/moco-tf/src/Annotations/StrideData.h @@ -0,0 +1,48 @@ +/* + * Copyright (c) 2019 Samsung Electronics Co., Ltd. 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. + */ + +#ifndef __MOCO_TF_STRIDE_DATA_H__ +#define __MOCO_TF_STRIDE_DATA_H__ + +#include + +namespace moco +{ +namespace tf +{ + +/** + * @brief StrideData holds temporary 'stride' attribute of TFConv2D, same thing + * like the one in Conv2D used for Canonicalizing TFConv2D to Conv2D. + * 'stride' will be calculated in FixShapeTransformation as for now. + */ +class StrideData : public loco::NodeAnnotation +{ +public: + StrideData() = default; + +public: + const loco::Stride<2> *stride(void) const { return &_stride; } + loco::Stride<2> *stride(void) { return &_stride; } + +private: + loco::Stride<2> _stride; +}; + +} // namespace tf +} // namespace moco + +#endif // __MOCO_TF_STRIDE_DATA_H__ -- 2.7.4