MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / DumpArray

Function DumpArray

tensorflow/lite/toco/dump_graphviz.cc:623–639  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

621}
622
623void DumpArray(const Model& model, string* output_file,
624 const string& array_id) {
625 Color color;
626 string shape;
627 GetArrayColorAndShape(model, array_id, &color, &shape);
628 string label = GetArrayLabel(model, array_id);
629 AppendF(output_file, kArrayNodeFmt, array_id, label, array_id, shape,
630 color.AsHexString(), color.TextColorString());
631
632 // Ops are placed in the same subgraph as their first output.
633 for (int op_index = 0; op_index < model.operators.size(); op_index++) {
634 const Operator& op = *model.operators[op_index];
635 if (!op.outputs.empty() && (op.outputs[0] == array_id)) {
636 DumpOperator(model, output_file, op_index);
637 }
638 }
639}
640
641void DumpNode(const Model& model, string* output_file, const string& node_name,
642 Node const& node) {

Callers 1

DumpNodeFunction · 0.85

Calls 8

GetArrayColorAndShapeFunction · 0.85
GetArrayLabelFunction · 0.85
AppendFFunction · 0.85
DumpOperatorFunction · 0.85
AsHexStringMethod · 0.80
TextColorStringMethod · 0.80
sizeMethod · 0.45
emptyMethod · 0.45

Tested by

no test coverage detected