diff options
Diffstat (limited to 'runtimes/neurun/src/dumper/dot/DotOperandInfo.h')
-rw-r--r-- | runtimes/neurun/src/dumper/dot/DotOperandInfo.h | 77 |
1 files changed, 77 insertions, 0 deletions
diff --git a/runtimes/neurun/src/dumper/dot/DotOperandInfo.h b/runtimes/neurun/src/dumper/dot/DotOperandInfo.h new file mode 100644 index 000000000..c54da444d --- /dev/null +++ b/runtimes/neurun/src/dumper/dot/DotOperandInfo.h @@ -0,0 +1,77 @@ +/* + * Copyright (c) 2018 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 __NEURUN_DUMPER_DOT_DOT_OPERAND_INFO_H__ +#define __NEURUN_DUMPER_DOT_DOT_OPERAND_INFO_H__ + +#include <vector> + +#include "IDotInfo.h" +#include "model/operand/Object.h" +#include "model/operand/Index.h" + +namespace neurun +{ +namespace dumper +{ +namespace dot +{ + +class DotOperandInfo : public IDotInfo +{ +public: + enum class Type + { + UNDEFINED, + MODEL_INPUT, + MODEL_OUTPUT, + INTERNAL + }; + +public: + static const std::string INPUT_SHAPE; + static const std::string OUTPUT_SHAPE; + static const std::string OPERAND_SHAPE; + static const std::string BG_COLOR_SCHEME; + static const std::string BG_COLORS[8]; + +public: + DotOperandInfo(const neurun::model::operand::Index &index, + const neurun::model::operand::Object &object, Type type); + +public: + virtual std::string index_str() const override; + virtual std::string label() const override; + virtual std::string dot_shape() const override; + virtual std::string bg_color_scheme() const override; + virtual std::string bg_color() const override; + +private: + void addBackendLabel(); + +private: + const neurun::model::operand::Index &_index; + const neurun::model::operand::Object &_object; + Type _type; + + std::vector<std::string> _labels; +}; + +} // namespace dot +} // namespace dumper +} // namespace neurun + +#endif // __NEURUN_DUMPER_DOT_DOT_OPERAND_INFO_H__ |