From 4f13335b34a5ad4fe4f7db1d5d65930b4e46e5c6 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=A2=85=ED=98=84/On-Device=20Lab=28SR=29/Staff?= =?utf8?q?=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Fri, 2 Aug 2019 12:59:26 +0900 Subject: [PATCH] [locop] Show Push/Pull index field (#6132) CanonicalNodeDescBuilder now shows the index attribute of Push/Pull nodes. Signed-off-by: Jonghyun Park --- compiler/locop/src/FormattedGraph.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/compiler/locop/src/FormattedGraph.cpp b/compiler/locop/src/FormattedGraph.cpp index b2d78f4..a855e40 100644 --- a/compiler/locop/src/FormattedGraph.cpp +++ b/compiler/locop/src/FormattedGraph.cpp @@ -238,6 +238,7 @@ public: { NodeDesc res{opname(node)}; + res.arg("index", node->indexed() ? pp::fmt(node->index()) : pp::fmt('?')); res.arg("from", symbol_lookup(*_symtbl, node->from())); res.arg("shape", pp::fmt(pretty(tensor_shape(node)))); res.state(NodeDesc::State::Complete); @@ -249,6 +250,7 @@ public: { NodeDesc res{opname(node)}; + res.arg("index", node->indexed() ? pp::fmt(node->index()) : pp::fmt('?')); // TODO Print dtype res.arg("shape", pp::fmt(pretty(tensor_shape(node)))); res.state(NodeDesc::State::PartiallyKnown); -- 2.7.4