84 ss <<
"FusedConvolutionBatchNormalizationNode";
92 ss <<
"FusedDepthwiseConvolutionBatchNormalizationNode";
105 std::stringstream
ss;
140 void DotGraphPrinter::print_header(
const Graph &g, std::ostream &os)
143 std::string graph_name = (g.
name().empty()) ?
"Graph" : g.
name();
144 os <<
"digraph " << graph_name <<
"{\n";
147 void DotGraphPrinter::print_footer(
const Graph &g, std::ostream &os)
153 void DotGraphPrinter::print_nodes(
const Graph &g, std::ostream &os)
155 for(
const auto &n : g.
nodes())
161 os << node_id <<
" ";
164 n->accept(_dot_node_visitor);
166 std::string
name = n->name().empty() ? node_id : n->name();
167 auto node_description = _dot_node_visitor.info();
169 os << R
"([label = ")" << name << R"( \n )" << n->assigned_target() << R"( \n )" << node_description << R"("])"; 175 void DotGraphPrinter::print_edges(
const Graph &g, std::ostream &os)
177 for(
const auto &e : g.
edges())
183 os << source_node_id <<
" -> " << sink_node_id <<
" ";
DataType data_type
Data type.
TensorShape shape
Tensor shape.
Batch Normalization Layer node.
Fused Depthwise Convolution Batch Normalization node.
Normalization Layer node.
void default_visit(INode &n) override
Function to be overloaded by the client and implement default behavior for the non-overloaded visitor...
bool enabled() const
Check if initialised.
DataLayoutDimension concatenation_axis() const
Concatenation axis parameter accessor.
const std::vector< std::unique_ptr< Edge > > & edges() const
Returns edges of graph.
NormType type() const
Get the normalization type.
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.
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.
Copyright (c) 2017-2021 Arm Limited.
Batch Normalization node.
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.
std::string to_string(const ROIPoolingLayerInfo &pool_info)
Formatted output of the ROIPoolingInfo type.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
bool is_enabled() const
Enabled parameter accessor.
void visit(ActivationLayerNode &n) override
Visit ActivationLayerNode.
Concatenation Layer node.
ConvolutionMethod convolution_method() const
Convolution layer method accessor.
const std::vector< NodeID > & nodes(NodeType type)
Returns graph input nodes.
PadStrideInfo pad_stride_info
ActivationLayerInfo activation_info() const
Activation metadata accessor.
Depthwise Convolution Layer node.
ActivationFunction activation() const
Get the type of activation function.
DataLayout layout
Data layout.