From 1e95ebc959f643e4eabc73fec3ba376ea5f2d405 Mon Sep 17 00:00:00 2001
From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princip?=
 =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?=
 <saehie.park@samsung.com>
Date: Mon, 10 Jun 2019 19:01:33 +0900
Subject: [PATCH] [moco/tf] const test with shape inference (#3709)

* [moco/tf] const test with shape inference

This will add test of ShapeInferenceData for ConstGen node

Signed-off-by: SaeHie Park <saehie.park@samsung.com>

* use name shape
---
 contrib/moco/lib/frontend/tf/src/Op/Const.test.cpp | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/contrib/moco/lib/frontend/tf/src/Op/Const.test.cpp b/contrib/moco/lib/frontend/tf/src/Op/Const.test.cpp
index 41da1fb..f444ff0 100644
--- a/contrib/moco/lib/frontend/tf/src/Op/Const.test.cpp
+++ b/contrib/moco/lib/frontend/tf/src/Op/Const.test.cpp
@@ -16,6 +16,8 @@
 
 #include "TestHelper.h"
 
+#include "Annotations/ShapeInferenceData.h"
+
 #include <moco/tf/Frontend.h>
 
 #include <loco.h>
@@ -96,6 +98,12 @@ TEST(TensorFlowFrontend, const_float_01)
   loco::Push *node1 = dynamic_cast<loco::Push *>(nodes->at(1));
   ASSERT_EQ(node1, push);
 
+  const auto *shape = node0->annot<moco::tf::ShapeInferenceData>();
+  ASSERT_NE(shape, nullptr);
+  ASSERT_EQ(shape->rank(), 2);
+  ASSERT_EQ(shape->dim(0), 2);
+  ASSERT_EQ(shape->dim(1), 3);
+
   ASSERT_EQ(node0->size<loco::DataType::FLOAT32>(), 6);
   ASSERT_EQ(node0->at<loco::DataType::FLOAT32>(0), 1.1f);
   ASSERT_EQ(node0->at<loco::DataType::FLOAT32>(1), 2.2f);
-- 
2.7.4