From: 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com>
Date: Wed, 26 Jun 2019 00:48:16 +0000 (+0900)
Subject: [moco_tf] add get_float_attr (#3973)
X-Git-Tag: nncc_backup~268
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=54604aeea33dec2a03827bb8e627a260eaeac2bf;p=platform%2Fcore%2Fml%2Fnnfw.git

[moco_tf] add get_float_attr (#3973)

This will add get_float_attr method for accessing attribute float value for tensorflow node

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

diff --git a/contrib/moco-tf/src/Convert.cpp b/contrib/moco-tf/src/Convert.cpp
index 03e5119..8232f2d 100644
--- a/contrib/moco-tf/src/Convert.cpp
+++ b/contrib/moco-tf/src/Convert.cpp
@@ -90,6 +90,14 @@ int64_t get_int_attr(const tensorflow::NodeDef &node, const std::string &attr_na
   return attr.i();
 }
 
+float get_float_attr(const tensorflow::NodeDef &node, const std::string &attr_name)
+{
+  assert(has_attr(node, attr_name));
+  const auto &attr = node.attr().at(attr_name);
+  assert(attr.value_case() == tensorflow::AttrValue::kF);
+  return attr.f();
+}
+
 loco::DataType as_loco_datatype(const tensorflow::DataType dtype)
 {
   switch (dtype)
diff --git a/contrib/moco-tf/src/Convert.h b/contrib/moco-tf/src/Convert.h
index 921fcff..83ba3c1 100644
--- a/contrib/moco-tf/src/Convert.h
+++ b/contrib/moco-tf/src/Convert.h
@@ -42,6 +42,7 @@ const tensorflow::AttrValue_ListValue &get_list_attr(const tensorflow::NodeDef &
                                                      const std::string &attr_name);
 const std::string &get_string_attr(const tensorflow::NodeDef &node, const std::string &attr_name);
 int64_t get_int_attr(const tensorflow::NodeDef &node, const std::string &attr_name);
+float get_float_attr(const tensorflow::NodeDef &node, const std::string &attr_name);
 
 loco::DataType as_loco_datatype(const tensorflow::DataType dtype);