From: 윤현식/On-Device Lab(SR)/Principal Engineer/삼성전자 Date: Wed, 31 Jul 2019 07:51:07 +0000 (+0900) Subject: [plier-tf] Moving moco test cases regarding data_layout (#6061) X-Git-Tag: submit/tizen/20190809.050447~296 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=b8833fdffe38378177c884ea7ab37b35382882b4;p=platform%2Fcore%2Fml%2Fnnfw.git [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 --- 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); +}