[moco/tf] simplify get_data_layout (#4124)

This will simplify get_data_layout to call as_DataLayout as common part

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 1cdf749..0e05a1a 100644 (file)
--- a/contrib/moco-tf/src/Convert.cpp
+++ b/contrib/moco-tf/src/Convert.cpp
@@ -154,13 +154,7 @@ const DataLayout get_data_layout(const tensorflow::NodeDef &node, const std::str
 {
   auto layout = get_string_attr(node, attr_name);
 
-  // TODO use as_DataLayout()
-  if (layout == "NHWC")
-    return moco::tf::DataLayout::NHWC;
-  else if (layout == "NCHW")
-    return moco::tf::DataLayout::NCHW;
-  else
-    throw std::runtime_error("unknown data layout");
+  return as_DataLayout(layout);
 }
 
 } // namespace tf