diff options
Diffstat (limited to 'compiler/luci/logex/src/FormattedGraph.cpp')
| -rw-r--r-- | compiler/luci/logex/src/FormattedGraph.cpp | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/compiler/luci/logex/src/FormattedGraph.cpp b/compiler/luci/logex/src/FormattedGraph.cpp index 4725ee3df..f04a418ef 100644 --- a/compiler/luci/logex/src/FormattedGraph.cpp +++ b/compiler/luci/logex/src/FormattedGraph.cpp @@ -244,6 +244,7 @@ private: IMPLEMENT(luci::CircleMirrorPad) IMPLEMENT(luci::CircleMul) IMPLEMENT(luci::CircleNeg) + IMPLEMENT(luci::CircleNonMaxSuppressionV4) IMPLEMENT(luci::CircleNotEqual) IMPLEMENT(luci::CircleOneHot) IMPLEMENT(luci::CirclePack) @@ -291,6 +292,7 @@ private: IMPLEMENT(luci::CircleTopKV2) IMPLEMENT(luci::CircleTranspose) IMPLEMENT(luci::CircleTransposeConv) + IMPLEMENT(luci::CircleUnique) IMPLEMENT(luci::CircleUnpack) IMPLEMENT(luci::CircleWhere) IMPLEMENT(luci::CircleWhile) @@ -303,9 +305,11 @@ private: IMPLEMENT(luci::CircleInput) IMPLEMENT(luci::CircleOutput) IMPLEMENT(luci::CircleIfOut) + IMPLEMENT(luci::CircleNonMaxSuppressionV4Out) IMPLEMENT(luci::CircleSplitOut) IMPLEMENT(luci::CircleSplitVOut) IMPLEMENT(luci::CircleTopKV2Out) + IMPLEMENT(luci::CircleUniqueOut) IMPLEMENT(luci::CircleUnpackOut) IMPLEMENT(luci::CircleWhileOut) #undef IMPLEMENT @@ -823,6 +827,19 @@ bool CircleNodeSummaryBuilder::summary(const luci::CircleNeg *node, locop::NodeS return use_x(tbl(), node, s); } +bool CircleNodeSummaryBuilder::summary(const luci::CircleNonMaxSuppressionV4 *node, + locop::NodeSummary &s) const +{ + s.args().append("boxes", pepper::str(node->boxes())); + s.args().append("scores", pepper::str(node->scores())); + s.args().append("max_output_size", pepper::str(node->max_output_size())); + s.args().append("iou_threshold", pepper::str(node->iou_threshold())); + s.args().append("score_threshold", pepper::str(node->score_threshold())); + + s.state(locop::NodeSummary::State::Complete); + return true; +} + bool CircleNodeSummaryBuilder::summary(const luci::CircleNotEqual *node, locop::NodeSummary &s) const { @@ -1227,6 +1244,14 @@ bool CircleNodeSummaryBuilder::summary(const luci::CircleTransposeConv *node, return true; } +bool CircleNodeSummaryBuilder::summary(const luci::CircleUnique *node, locop::NodeSummary &s) const +{ + s.args().append("input", tbl()->lookup(node->input())); + s.args().append("idx_out_type", to_str(node->idx_out_type())); + s.state(locop::NodeSummary::State::Complete); + return true; +} + bool CircleNodeSummaryBuilder::summary(const luci::CircleUnpack *node, locop::NodeSummary &s) const { s.args().append("value", tbl()->lookup(node->value())); @@ -1293,6 +1318,16 @@ bool CircleNodeSummaryBuilder::summary(const luci::CircleTopKV2Out *node, return true; } +bool CircleNodeSummaryBuilder::summary(const luci::CircleUniqueOut *node, + locop::NodeSummary &s) const +{ + s.args().append("unique", tbl()->lookup(node->input())); + + s.state(locop::NodeSummary::State::Complete); + + return true; +} + bool CircleNodeSummaryBuilder::summary(const luci::CircleUnpackOut *node, locop::NodeSummary &s) const { @@ -1308,6 +1343,12 @@ bool CircleNodeSummaryBuilder::summary(const luci::CircleIfOut *node, locop::Nod return use_input(tbl(), node, s); } +bool CircleNodeSummaryBuilder::summary(const luci::CircleNonMaxSuppressionV4Out *node, + locop::NodeSummary &s) const +{ + return use_input(tbl(), node, s); +} + bool CircleNodeSummaryBuilder::summary(const luci::CircleWhileOut *node, locop::NodeSummary &s) const { |