[moco-tf] set shape in Frontend test (#8408)
author채성우/On-Device Lab(SR)/Engineer/삼성전자 <sw4670.chae@samsung.com>
Wed, 23 Oct 2019 05:35:32 +0000 (14:35 +0900)
committer박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Wed, 23 Oct 2019 05:35:32 +0000 (14:35 +0900)
* [moco-tf] set shape in Frontend test

This commit set shape to signature in Frontend test.

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

compiler/moco-tf/src/Frontend.test.cpp

index 57a9eb7..561d909 100644 (file)
@@ -63,6 +63,7 @@ TEST(FrontendTests, testcase_000)
   moco::tf::ModelSignature signature;
 
   signature.add_input(moco::tf::TensorName("Placeholder", 0));
+  signature.shape("Placeholder:0", angkor::TensorShape{4});
   signature.add_output(moco::tf::TensorName("Identity", 0));
 
   std::stringstream ss{pbtxt_000};