From 2e836ca76822050b7568aa1f68939a4d4917c061 Mon Sep 17 00:00:00 2001
From: Haibin Lin <linhaibin.eric@gmail.com>
Date: Thu, 5 Apr 2018 13:55:59 -0700
Subject: [PATCH] declare type name for optional<TShape> (#429)
* declare type for optional tshape
* add doc
* move code to another place
---
nnvm/include/nnvm/tuple.h | 2 ++
1 file changed, 2 insertions(+)
diff --git a/nnvm/include/nnvm/tuple.h b/nnvm/include/nnvm/tuple.h
index 353b79ab0..c5caa9bd8 100644
--- a/nnvm/include/nnvm/tuple.h
+++ b/nnvm/include/nnvm/tuple.h
@@ -611,6 +611,8 @@ struct hash<nnvm::TShape> {
} // namespace std
namespace dmlc {
+/*! \brief description for optional TShape */
+DMLC_DECLARE_TYPE_NAME(optional<nnvm::TShape>, "Shape or None");
// avoid low version of MSVC
#if !defined(_MSC_VER)
template<typename T>
--
GitLab