From b16f2d9a9dfb959f627afe92339c8fecb4bc4f53 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=B1=84=EC=84=B1=EC=9A=B0/On-Device=20Lab=28SR=29/Enginee?= =?utf8?q?r/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Fri, 18 Oct 2019 16:24:39 +0900 Subject: [PATCH] [moco-tf] fix PadCanonicalizer (#8298) * [moco-tf] fix PadCanonicalizer This commit fixs PadCanonicalizer in moco-tf. Signed-off-by: seongwoo * add comment. --- compiler/moco-tf/src/Canonicalization/PadCanonicalizer.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/compiler/moco-tf/src/Canonicalization/PadCanonicalizer.cpp b/compiler/moco-tf/src/Canonicalization/PadCanonicalizer.cpp index 7f9db61..4296273 100644 --- a/compiler/moco-tf/src/Canonicalization/PadCanonicalizer.cpp +++ b/compiler/moco-tf/src/Canonicalization/PadCanonicalizer.cpp @@ -56,7 +56,8 @@ bool canonicalize_pad(loco::Graph *graph, moco::tf::TFPad *node) // TODO: support other dtype. assert(loco::dtype_get(input_node) == loco::DataType::FLOAT32); constant_node->dtype(loco::DataType::FLOAT32); - constant_node->rank(0); + // TODO: constant node changes to scalar when it is implemented. + constant_node->shape({1}); constant_node->size(1); constant_node->at(0) = 0.0f; -- 2.7.4