From: 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 Date: Tue, 9 Jul 2019 22:07:06 +0000 (+0900) Subject: [moco/tf] Introduce StrideData (#4158) X-Git-Tag: nncc_backup~143 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=a88b1aa2be95e55075b6a08ce570a6ff638a99db;p=platform%2Fcore%2Fml%2Fnnfw.git [moco/tf] Introduce StrideData (#4158) This will introduce StrideData used as annotation for TFConv2D stride Signed-off-by: SaeHie Park --- 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__