[moco-tf] fix PadCanonicalizer (#8298)
author채성우/On-Device Lab(SR)/Engineer/삼성전자 <sw4670.chae@samsung.com>
Fri, 18 Oct 2019 07:24:39 +0000 (16:24 +0900)
committer박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Fri, 18 Oct 2019 07:24:39 +0000 (16:24 +0900)
* [moco-tf] fix PadCanonicalizer

This commit fixs PadCanonicalizer in moco-tf.

Signed-off-by: seongwoo <sw4670.chae@samsung.com>
* add comment.

compiler/moco-tf/src/Canonicalization/PadCanonicalizer.cpp

index 7f9db61..4296273 100644 (file)
@@ -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<loco::DataType::FLOAT32>(1);
   constant_node->at<loco::DataType::FLOAT32>(0) = 0.0f;