declare a type name for each tuple type (#151)

* declare a type name for each tuple type * generic way to declare type names for each tuple type * fix lint error * update submodule dmlc-core
eric-haibin-lin · Sep 24, 2017 · 8081a7f · 8081a7f
1 parent e75c53e
commit 8081a7f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/dmlc-core b/dmlc-core
diff --git a/include/nnvm/tuple.h b/include/nnvm/tuple.h
@@ -11,6 +11,7 @@
 #include <algorithm>
 #include <utility>
 #include <iostream>
+#include <string>
 #include "./base.h"
 
 namespace nnvm {
@@ -611,4 +612,13 @@ struct hash<nnvm::TShape> {
 };
 }  // namespace std
 
+namespace dmlc {
+template<typename T>
+struct type_name_helper<nnvm::Tuple<T> > {
+  static inline std::string value() {
+    return "tuple of <" + type_name<T>() + ">";
+  }
+};
+}
+
 #endif  // NNVM_TUPLE_H_