From: 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 Date: Mon, 3 Jun 2019 01:23:22 +0000 (+0900) Subject: [moco/tf] Introduce ConcatData (#3641) X-Git-Tag: nncc_backup~495 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=6cf1518ed020d8395f2f72dad3cb12c7279133e9;p=platform%2Fcore%2Fml%2Fnnfw.git [moco/tf] Introduce ConcatData (#3641) * [moco/tf] Introduce ConcatData This will introduce ConcatData used for annotation of Concat node. Signed-off-by: SaeHie Park * fix --- diff --git a/contrib/moco/lib/frontend/tf/src/Annotations/ConcatData.h b/contrib/moco/lib/frontend/tf/src/Annotations/ConcatData.h new file mode 100644 index 0000000..4c8e5fa --- /dev/null +++ b/contrib/moco/lib/frontend/tf/src/Annotations/ConcatData.h @@ -0,0 +1,44 @@ +/* + * 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_CONCAT_DATA_H__ +#define __MOCO_TF_CONCAT_DATA_H__ + +#include + +namespace moco +{ +namespace tf +{ + +/** + * @brief ConcatData holds temporary axis attribute of Concat while building the graph +*/ +class ConcatData : public loco::NodeAnnotation +{ +public: + ConcatData(int32_t axis) : _axis(axis) {} + + int32_t axis(void) const { return _axis; } + +private: + int32_t _axis; +}; + +} // namespace tf +} // namespace moco + +#endif // __MOCO_TF_CONCAT_DATA_H__