runtime/onert/core/src/dumper/dot/DotSubgraphInfo.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/Operations.h>
  24 #include "ir/OpSequence.h"
  25 #include "util/Set.h"
  26
  27 namespace onert
  28 {
  29 namespace dumper
  30 {
  31 namespace dot
  32 {
  33
  34 class DotSubgraphInfo
  35 {
  36 public:
  37   DotSubgraphInfo(const ir::OpSequenceIndex &index, const ir::OpSequence &op_seq,
  38                   const util::Set<ir::OperandIndex> &shown_operands,
  39                   const ir::Operations &operations_ctx);
  40
  41   ir::OpSequenceIndex index() const { return _index; }
  42   std::string label() const { return _label; }
  43   void label(const std::string &val) { _label = val; }
  44   std::string fillcolor() const { return _fillcolor; }
  45   void fillcolor(const std::string &val) { _fillcolor = val; }
  46   const std::unordered_set<ir::OperationIndex> &operations() const { return _operations; }
  47   const std::unordered_set<ir::OperandIndex> &operands() const { return _operands; }
  48
  49 private:
  50   ir::OpSequenceIndex _index;
  51   std::string _label;
  52   std::string _fillcolor;
  53   std::unordered_set<ir::OperationIndex> _operations;
  54   std::unordered_set<ir::OperandIndex> _operands;
  55 };
  56
  57 } // namespace dot
  58 } // namespace dumper
  59 } // namespace onert
  60
  61 #endif // __ONERT_CORE_DUMPER_DOT_DOT_SUBGRAPH_INFO_H__