DotBuilder.DotBuilder Class Reference

Public Member Functions
	__init__ (self, str circle_path, str dot_path, str metric, str colors)
	save (self, dict qerror_map)

Protected Member Functions
	_get_color (self, float qerror)
	_gen_color_table (self)

Protected Attributes
	_model
	_name
	_dot_path
	_metric
	_colors

Detailed Description

Definition at line 34 of file DotBuilder.py.

Constructor & Destructor Documentation

__init__()

DotBuilder.DotBuilder.__init__	(		self,
		str	circle_path,
		str	dot_path,
		str	metric,
		str	colors
	)

circle_path: Path to the fp32 circle model (required to build graph)
dot_path: Path to the saved dot file
metric: Metric name (ex: MPEIR, MSE)
colors: List of color slots [{'b': begin, 'e': end, 'c':color}, ..]

Definition at line 35 of file DotBuilder.py.

    def __init__(self, circle_path: str, dot_path: str, metric: str, colors: str):
        '''
        circle_path: Path to the fp32 circle model (required to build graph)
        dot_path: Path to the saved dot file
        metric: Metric name (ex: MPEIR, MSE)
        colors: List of color slots [{'b': begin, 'e': end, 'c':color}, ..]
        '''
        with open(circle_path, 'rb') as f:
            self._model = Model.Model.GetRootAsModel(f.read())
 
        if self._model.SubgraphsLength() != 1:
            raise RuntimeError("Only one subgraph is supported")
 
        self._name = Path(circle_path).name
        self._dot_path = dot_path
        self._metric = metric
        self._colors = colors
 

Member Function Documentation

_gen_color_table()

DotBuilder.DotBuilder._gen_color_table (   self )

protected

Definition at line 74 of file DotBuilder.py.

    def _gen_color_table(self):
        color_table = "< <table>"
        for slot in self._colors:
            begin = slot['b']
            end = slot['e']
            color = slot['c']
            color_table += "<tr> <td bgcolor=\""
            color_table += color
            color_table += "\">"
            color_table += self._metric + ": {:.4f}".format(
                begin) + " ~ " + "{:.4f}".format(end)
            color_table += "</td> </tr>"
        color_table += "</table> >"
        return pydot.Node("color_table", shape='none', label=color_table)
 

References DotBuilder.DotBuilder._colors, Palette.Palette._colors, Palette.UniformPalette._colors, mpqsolver::core::DatasetEvaluator._metric, and DotBuilder.DotBuilder._metric.

Referenced by DotBuilder.DotBuilder.save().

_get_color()

DotBuilder.DotBuilder._get_color (   self, float  qerror  )

protected

Definition at line 54 of file DotBuilder.py.

    def _get_color(self, qerror: float):
        # Find a slot where qerror is in the range of [begin, end]
        for slot in self._colors:
            begin = slot['b']
            end = slot['e']
            if (qerror > begin or math.isclose(
                    qerror, begin)) and (qerror < end or math.isclose(qerror, end)):
                return slot['c']
 
        # Use the first color if qerror is smaller than the first begin
        if qerror < self._colors[0]['b']:
            return self._colors[0]['c']
 
        # Use the last color if qerror is larger than the last end
        if qerror > self._colors[-1]['e']:
            return self._colors[-1]['c']
 
        raise RuntimeError("Color ID not found. QError: " + str(qerror))
 

References DotBuilder.DotBuilder._colors, Palette.Palette._colors, and Palette.UniformPalette._colors.

Referenced by DotBuilder.DotBuilder.save().

save()

DotBuilder.DotBuilder.save	(		self,
		dict	qerror_map
	)

qerror_map: Dictionary of {op_name (str) -> qerror (float)}

Definition at line 90 of file DotBuilder.py.

    def save(self, qerror_map: dict):
        '''
        qerror_map: Dictionary of {op_name (str) -> qerror (float)}
        '''
        # Build graph
        DOT = pydot.Dot(self._name, graph_type="digraph")
 
        # Add color table
        DOT.add_node(self._gen_color_table())
 
        # Dictionary from output tensor name to Op name {str -> str}
        # This dict is for handling Ops with multiple output tensors.
        # We use the first output tensor's name as the Op name, following
        # the implementation of luci IR
        output_to_op = dict()
 
        graph = self._model.Subgraphs(0)
 
        # Add Input nodes
        for i in range(graph.InputsLength()):
            name = _tensor_name(graph, graph.Inputs(i))
            output_to_op[name] = name
            DOT.add_node(pydot.Node(_quote(name)))
 
        # Add Output nodes
        for i in range(graph.OutputsLength()):
            name = _tensor_name(graph, graph.Outputs(i))
            output_to_op[name] = name
            DOT.add_node(pydot.Node(_quote(name)))
 
        # Add Edges
        for i in range(graph.OperatorsLength()):
            op = graph.Operators(i)
            # Name of the first output tensor
            op_name = _tensor_name(graph, op.Outputs(0))
            if op.OutputsLength() == 0:
                print(op_name)
                continue
 
            if op_name in qerror_map:
                qerror = qerror_map[op_name]
                node = pydot.Node(_quote(op_name),
                                  style="filled",
                                  fillcolor=self._get_color(qerror),
                                  xlabel=self._metric + ": {:.4f}".format(qerror))
            else:
                # qerror_map does not have qerror info for the op. Color gray.
                # When this happen? visq does not collect qerror info of some Ops
                # For example, Reshape Op does not change values, so its qerror
                # info is not collected.
                node = pydot.Node(_quote(op_name), style="filled", fillcolor='gray')
 
            DOT.add_node(node)
 
            for output_idx in range(op.OutputsLength()):
                output_name = _tensor_name(graph, op.Outputs(output_idx))
                # Set Op name as the first output tensor name (op_name)
                output_to_op[output_name] = op_name
 
            for j in range(op.InputsLength()):
                op_input = op.Inputs(j)
 
                # Optional input case (ex: For TConv with no bias, bias is -1)
                if op_input == -1:
                    continue
 
                op_input_name = _tensor_name(graph, op_input)
                if op_input_name not in output_to_op:
                    continue
 
                # Use the saved name to handle multiple outputs
                op_input_name = output_to_op[op_input_name]
                DOT.add_edge(pydot.Edge(_quote(op_input_name), _quote(op_name)))
 
        DOT.write(self._dot_path)

References DotBuilder.DotBuilder._dot_path, DotBuilder.DotBuilder._gen_color_table(), DotBuilder.DotBuilder._get_color(), mpqsolver::core::DatasetEvaluator._metric, DotBuilder.DotBuilder._metric, SingleOperatorTest.SingleOperatorTest._model, ScalarOperandDecl._model, TensorOperandDecl._model, WeightDecl._model, OperationDecl._model, luci::CircleReader._model, mir_onnx::ONNXImporterImpl._model, mir_tflite::TfliteImporter._model, GenericBackend._model, onnx_legalizer._ModelTransformerHelper._model, DotBuilder.DotBuilder._model, luci_interpreter::CircleReader._model, onert_micro::core::reader::OMCircleReader._model, onert::compiler::Compiler._model, onert::exporter::CircleExporter._model, onert::compiler::train::TrainingCompiler._model, onert::loader::BaseLoader< LoaderDomain >._model, cli::App._name, coco::Arg._name, enco::Pass._name, TFLTensorInfo._name, loco::NamedEntity._name, locoex::COpCall._name, locop::NodeDesc._name, luci_interpreter::Tensor._name, CircleTensorInfo._name, luci::SubGraphContext._name, luci::CircleNode._name, UnrollLSTM._name, luci::CircleNodeOrigin::Source._name, mir::Operation::Output._name, TFReluGraphUpdate._name, TFRelu6GraphUpdate._name, TFRsqrtGraphUpdate._name, TFSqrtGraphUpdate._name, TFStopGradientGraphUpdate._name, TFTanhGraphUpdate._name, moco::TFNode._name, moco::TensorName._name, nnc::ArtifactNamed._name, nnc::ArtifactModule._name, DotBuilder.DotBuilder._name, DotBuilder._quote(), and DotBuilder._tensor_name().

Field Documentation

_colors

DotBuilder.DotBuilder._colors

protected

Definition at line 51 of file DotBuilder.py.

Referenced by DotBuilder.DotBuilder._gen_color_table(), DotBuilder.DotBuilder._get_color(), and Palette.Palette.colorscheme().

_dot_path

DotBuilder.DotBuilder._dot_path

protected

Definition at line 49 of file DotBuilder.py.

Referenced by DotBuilder.DotBuilder.save().

_metric

DotBuilder.DotBuilder._metric

protected

Definition at line 50 of file DotBuilder.py.

Referenced by DotBuilder.DotBuilder._gen_color_table(), and DotBuilder.DotBuilder.save().

_model

DotBuilder.DotBuilder._model

protected

Definition at line 43 of file DotBuilder.py.

Referenced by onnx_legalizer._ModelTransformerHelper.delete_marked_nodes(), onnx_legalizer._ModelTransformerHelper.make_constant_tensor(), onnx_legalizer._ModelTransformerHelper.make_node(), and DotBuilder.DotBuilder.save().

_name

DotBuilder.DotBuilder._name

protected

Definition at line 48 of file DotBuilder.py.

Referenced by DotBuilder.DotBuilder.save().

---

The documentation for this class was generated from the following file:

• compiler/visq/visqlib/DotBuilder.py