Compute Library
 21.02
DotGraphPrinter.cpp
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2018-2020 Arm Limited.
3  *
4  * SPDX-License-Identifier: MIT
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to
8  * deal in the Software without restriction, including without limitation the
9  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10  * sell copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in all
14  * copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
25 
26 #include "arm_compute/core/Error.h"
31 
32 namespace arm_compute
33 {
34 namespace graph
35 {
37 {
38  std::stringstream ss;
39  ss << n.activation_info().activation();
40  _info = ss.str();
41 }
42 
44 {
45  std::stringstream ss;
46  ss << (n.fused_activation().enabled() ? to_string(n.fused_activation().activation()) : "");
47  _info = ss.str();
48 }
49 
51 {
52  std::stringstream ss;
53  ss << "Enabled: " << n.is_enabled();
54  ss << R"( \n )";
55  ss << "Axis: " << n.concatenation_axis();
56  _info = ss.str();
57 }
58 
60 {
61  std::stringstream ss;
62  ss << n.convolution_method();
63  _info = ss.str();
64 }
65 
67 {
68  std::stringstream ss;
70  _info = ss.str();
71 }
72 
74 {
75  std::stringstream ss;
76  ss << n.eltwise_operation();
77  _info = ss.str();
78 }
79 
81 {
83  std::stringstream ss;
84  ss << "FusedConvolutionBatchNormalizationNode";
85  _info = ss.str();
86 }
87 
89 {
91  std::stringstream ss;
92  ss << "FusedDepthwiseConvolutionBatchNormalizationNode";
93  _info = ss.str();
94 }
95 
97 {
98  std::stringstream ss;
99  ss << n.normalization_info().type();
100  _info = ss.str();
101 }
102 
104 {
105  std::stringstream ss;
106  ss << n.pooling_info().pool_type;
107  ss << R"( \n )";
108  ss << n.pooling_info().pool_size;
109  ss << R"( \n )";
110  ss << n.pooling_info().pad_stride_info;
111  _info = ss.str();
112 }
113 
115 {
116  _info.clear();
117 }
118 
119 const std::string &DotGraphVisitor::info() const
120 {
121  return _info;
122 }
123 
124 void DotGraphPrinter::print(const Graph &g, std::ostream &os)
125 {
126  // Print header
127  print_header(g, os);
128 
129  // Print nodes
130  print_nodes(g, os);
131 
132  // Print edges
133  print_edges(g, os);
134 
135  // Print footer
136  print_footer(g, os);
137 }
138 
139 void DotGraphPrinter::print_header(const Graph &g, std::ostream &os)
140 {
141  // Print graph name
142  std::string graph_name = (g.name().empty()) ? "Graph" : g.name();
143  os << "digraph " << graph_name << "{\n";
144 }
145 
146 void DotGraphPrinter::print_footer(const Graph &g, std::ostream &os)
147 {
149  os << "}\n";
150 }
151 
152 void DotGraphPrinter::print_nodes(const Graph &g, std::ostream &os)
153 {
154  for(const auto &n : g.nodes())
155  {
156  if(n)
157  {
158  // Output node id
159  std::string node_id = std::string("n") + support::cpp11::to_string(n->id());
160  os << node_id << " ";
161 
162  // Output label
163  n->accept(_dot_node_visitor);
164 
165  std::string name = n->name().empty() ? node_id : n->name();
166  auto node_description = _dot_node_visitor.info();
167 
168  os << R"([label = ")" << name << R"( \n )" << n->assigned_target() << R"( \n )" << node_description << R"("])";
169  os << ";\n";
170  }
171  }
172 }
173 
174 void DotGraphPrinter::print_edges(const Graph &g, std::ostream &os)
175 {
176  for(const auto &e : g.edges())
177  {
178  if(e)
179  {
180  std::string source_node_id = std::string("n") + support::cpp11::to_string(e->producer_id());
181  std::string sink_node_id = std::string("n") + support::cpp11::to_string(e->consumer_id());
182  os << source_node_id << " -> " << sink_node_id << " ";
183  const Tensor *t = e->tensor();
184  ARM_COMPUTE_ERROR_ON(t == nullptr);
185  os << R"([label = ")" << t->desc().shape << R"( \n )" << t->desc().data_type << R"( \n )" << t->desc().layout << R"("])";
186  os << ";\n";
187  }
188  }
189 }
190 } // namespace graph
191 } // namespace arm_compute
bool enabled() const
Check if initialised.
Definition: Types.h:1600
DataLayoutDimension concatenation_axis() const
Concatenation axis parameter accessor.
const std::vector< std::unique_ptr< Edge > > & edges() const
Returns edges of graph.
Definition: Graph.cpp:189
NormType type() const
Get the normalization type.
Definition: Types.h:1665
std::string to_string(T &&value)
Convert integer and float values to string.
std::stringstream ss(mlgo_str)
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
Definition: Error.h:466
DepthwiseConvolutionMethod depthwise_convolution_method() const
Depthwise convolution layer method accessor.
ActivationLayerInfo fused_activation() const
Returns fused activation.
EltwiseOperation eltwise_operation() const
Eltwise operation accessor.
const std::string & info() const
Returns the output information of the last visited node.
std::string name() const
Returns graph name.
Definition: Graph.cpp:164
Copyright (c) 2017-2021 Arm Limited.
PoolingLayerInfo pooling_info() const
Pooling metadata accessor.
void print(const Graph &g, std::ostream &os) override
Print graph.
NormalizationLayerInfo normalization_info() const
Normalization info accessor.
TensorDescriptor & desc()
TensorInfo metadata accessor.
Definition: Tensor.cpp:40
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
Definition: Error.h:152
bool is_enabled() const
Enabled parameter accessor.
void visit(ActivationLayerNode &n) override
Visit ActivationLayerNode.
const char * name
ConvolutionMethod convolution_method() const
Convolution layer method accessor.
Graph class.
Definition: Graph.h:53
const std::vector< NodeID > & nodes(NodeType type)
Returns graph input nodes.
Definition: Graph.cpp:174
PadStrideInfo pad_stride_info
Definition: Types.h:1302
std::string to_string(const ICLTensor &arg)
ActivationLayerInfo activation_info() const
Activation metadata accessor.
void default_visit() override
Function to be overloaded by the client and implement default behavior for the non-overloaded visitor...
ActivationFunction activation() const
Get the type of activation function.
Definition: Types.h:1585
Tensor object.
Definition: Tensor.h:41