This commit is contained in:
Chang Lan 2018-12-06 12:33:00 -08:00 коммит произвёл Tianqi Chen
Родитель d1a0c90106
Коммит 1ba443a4fe
3 изменённых файлов: 5 добавлений и 2 удалений

2
3rdparty/dmlc-core поставляемый

@ -1 +1 @@
Subproject commit 4d49691f1a9d944c3b0aa5e63f1db3cad1f941f8
Subproject commit 519d013a213c0c447a971f51219473ef564d2348

Просмотреть файл

@ -25,6 +25,9 @@ using dmlc::array_view;
/*!\brief getter function of any type */
using dmlc::get;
/*!\brief "unsafe" getter function of any type */
using dmlc::unsafe_get;
} // namespace nnvm
// describe op registration point

Просмотреть файл

@ -229,7 +229,7 @@ inline const T& Graph::GetAttr(const std::string& attr_name) const {
auto it = attrs.find(attr_name);
CHECK(it != attrs.end())
<< "Cannot find attribute " << attr_name << " in the graph";
return nnvm::get<T>(*it->second);
return nnvm::unsafe_get<T>(*it->second);
}
inline bool Graph::HasAttr(const std::string& attr_name) const {