84 ss <<
"FusedConvolutionBatchNormalizationNode";
92 ss <<
"FusedConvolutionBatchNormalizationWithPostOpsNode";
100 ss <<
"FusedConvolutionWithPostOpNode";
107 std::stringstream
ss;
108 ss <<
"FusedDepthwiseConvolutionBatchNormalizationNode";
114 std::stringstream
ss;
121 std::stringstream
ss;
156 void DotGraphPrinter::print_header(
const Graph &g, std::ostream &os)
159 std::string graph_name = (g.
name().empty()) ?
"Graph" : g.
name();
160 os <<
"digraph " << graph_name <<
"{\n";
163 void DotGraphPrinter::print_footer(
const Graph &g, std::ostream &os)
169 void DotGraphPrinter::print_nodes(
const Graph &g, std::ostream &os)
171 for(
const auto &
n : g.
nodes())
177 os << node_id <<
" ";
180 n->accept(_dot_node_visitor);
182 std::string
name =
n->name().empty() ? node_id :
n->name();
183 auto node_description = _dot_node_visitor.info();
185 os << R
"([label = ")" << name << R"( \n )" << n->assigned_target() << R"( \n )" << node_description << R"("])"; 191 void DotGraphPrinter::print_edges(
const Graph &g, std::ostream &os)
193 for(
const auto &e : g.
edges())
199 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.
#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.
std::string to_string(const T &val)
Fallback method: try to use std::to_string:
Batch Normalization 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.