runtime/onert/core/src/dumper/dot/DotOpSequenceInfo.h

   1 /*
   2  * Copyright (c) 2019 Samsung Electronics Co., Ltd. All Rights Reserved
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #ifndef __ONERT_CORE_DUMPER_DOT_DOT_SUBGRAPH_INFO_H__
  18 #define __ONERT_CORE_DUMPER_DOT_DOT_SUBGRAPH_INFO_H__
  19
  20 #include <unordered_set>
  21
  22 #include "ir/Index.h"
  23 #include "ir/OpSequence.h"
  24 #include "util/Set.h"
  25
  26 namespace onert
  27 {
  28 namespace dumper
  29 {
  30 namespace dot
  31 {
  32
  33 class DotOpSequenceInfo
  34 {
  35 public:
  36   DotOpSequenceInfo(const ir::OpSequenceIndex &index, const ir::OpSequence &op_seq,
  37                     const util::Set<ir::OperandIndex> &shown_operands);
  38
  39   ir::OpSequenceIndex index() const { return _index; }
  40   std::string label() const { return _label; }
  41   void label(const std::string &val) { _label = val; }
  42   std::string fillcolor() const { return _fillcolor; }
  43   void fillcolor(const std::string &val) { _fillcolor = val; }
  44   const std::unordered_set<ir::OperationIndex> &operations() const { return _operations; }
  45   const std::unordered_set<ir::OperandIndex> &operands() const { return _operands; }
  46
  47 private:
  48   ir::OpSequenceIndex _index;
  49   std::string _label;
  50   std::string _fillcolor;
  51   std::unordered_set<ir::OperationIndex> _operations;
  52   std::unordered_set<ir::OperandIndex> _operands;
  53 };
  54
  55 } // namespace dot
  56 } // namespace dumper
  57 } // namespace onert
  58
  59 #endif // __ONERT_CORE_DUMPER_DOT_DOT_SUBGRAPH_INFO_H__