[moco/tf] TensorName class to represent the name of tensor (#3776)

* [moco/tf] TensorName class to represent the name of tensor

This class represents tensor name, which is a form of _nodeName_ ":" _number

Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>
* adds a empty line

diff --git a/contrib/moco/lib/frontend/tf/include/moco/tf/Names.h b/contrib/moco/lib/frontend/tf/include/moco/tf/Names.h
new file mode 100644 (file)
index 0000000..fc612d2
--- /dev/null
+++ b/contrib/moco/lib/frontend/tf/include/moco/tf/Names.h
@@ -0,0 +1,93 @@
+/*
+ * Copyright (c) 2019 Samsung Electronics Co., Ltd. All Rights Reserved
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#ifndef __MOCO_TF_NAMES_H__
+#define __MOCO_TF_NAMES_H__
+
+#include <string>
+#include <stdexcept>
+
+namespace moco
+{
+namespace tf
+{
+
+struct TensorName final
+{
+public:
+  /**
+   * @brief Constructor
+   *
+   * @note If tensor_name does not have ":index", this constructor adds ":0" by default
+   */
+  explicit TensorName(const std::string &tensor_name)
+  {
+    if (tensor_name.find(":") != std::string::npos) // tensor_name is a form of letter:0
+    {
+      _name.assign(tensor_name);
+    }
+    else
+    {
+      _name.assign(tensor_name + ":0"); // if it does not have ":index", adds ":0" by default
+    }
+  }
+
+  explicit TensorName(const std::string &node_name, const int tensor_index)
+  {
+    if (node_name.find(":") != std::string::npos) // tensor_name is already a form of name:0
+    {
+      throw std::runtime_error("Node name has already tensor index:" + node_name);
+    }
+    else
+    {
+      _name.assign(node_name + ":" + std::to_string(tensor_index));
+    }
+  }
+
+  const std::string &name() const { return _name; }
+
+  /**
+   * @brief Returns node name from tensor name by removing, e.g., ":0"
+   */
+  const std::string nodeName() const
+  {
+    auto index = _name.find(":");
+
+    if (index != std::string::npos)
+      return _name.substr(0, index);
+    else
+      throw std::runtime_error{"Tensor name should be a name:number format: " + _name};
+  };
+
+private:
+  std::string _name;
+};
+
+/**
+ * @brief To use TensorName as a key in std::map, this struct defines how to compare two TensorNames
+ */
+struct TensorNameCompare
+{
+  bool operator()(const TensorName &lhs, const TensorName &rhs) const
+  {
+    return lhs.name() < rhs.name();
+  }
+};
+
+} // namespace tf
+} // namespace moco
+
+#endif // __MOCO_TF_NAMES_H__