[tf2tflite] Fix build break (#3786)
author박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Fri, 14 Jun 2019 06:41:19 +0000 (15:41 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Fri, 14 Jun 2019 06:41:19 +0000 (15:41 +0900)
Recent commit on moco changes the frontend interface, but this change is
not reflected into tf2tflite.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
contrib/tf2tflite/src/Driver.cpp

index bca0e35..92e9d30 100644 (file)
@@ -41,11 +41,11 @@ int main(int argc, char **argv)
       switch (info->kind())
       {
       case nnkit::support::tftestinfo::ParsedTensor::Kind::Input:
-        sig.add_input(info->nodeName());
+        sig.add_input(moco::tf::TensorName{info->name()});
         break;
 
       case nnkit::support::tftestinfo::ParsedTensor::Kind::Output:
-        sig.add_output(info->nodeName());
+        sig.add_output(moco::tf::TensorName{info->name()});
         break;
 
       default: