From b8833fdffe38378177c884ea7ab37b35382882b4 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EC=9C=A4=ED=98=84=EC=8B=9D/On-Device=20Lab=28SR=29/Princip?= =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Wed, 31 Jul 2019 16:51:07 +0900 Subject: [PATCH] [plier-tf] Moving moco test cases regarding data_layout (#6061) Test cases of moco were moved into plier. These test cases are for two methods that were moved lately from moco to plier. Signed-off-by: Hyun Sik Yoon --- compiler/plier-tf/src/Convert.test.cpp | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/compiler/plier-tf/src/Convert.test.cpp b/compiler/plier-tf/src/Convert.test.cpp index c7ca25a..a543db0 100644 --- a/compiler/plier-tf/src/Convert.test.cpp +++ b/compiler/plier-tf/src/Convert.test.cpp @@ -79,6 +79,11 @@ TEST(plier_Convert, attr_shape) ASSERT_EQ(shape.dim(3).size(), 8); } +TEST(plier_Convert, to_loco_datatype) +{ + ASSERT_EQ(plier::tf::as_loco_datatype(tensorflow::DT_FLOAT), loco::DataType::FLOAT32); +} + TEST(plier_Convert, attr_ilist) { tensorflow::NodeDef node; @@ -92,3 +97,9 @@ TEST(plier_Convert, attr_ilist) ASSERT_EQ(i_list.at(2), 1LL << 40); ASSERT_EQ(i_list.at(3), -(1LL << 40)); } + +TEST(plier_Convert, to_data_layout) +{ + ASSERT_EQ(plier::tf::as_data_layout("NHWC"), plier::tf::DataLayout::NHWC); + ASSERT_EQ(plier::tf::as_data_layout("NCHW"), plier::tf::DataLayout::NCHW); +} -- 2.7.4