[Bugfix] Missing headers (#3392)

apache · Jun 19, 2019 · 947131c · 947131c
1 parent 05c7728
commit 947131c
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 0 deletions.
diff --git a/include/tvm/relay/attrs/algorithm.h b/include/tvm/relay/attrs/algorithm.h
@@ -25,6 +25,7 @@
 #define TVM_RELAY_ATTRS_ALGORITHM_H_
 
 #include <tvm/attrs.h>
+#include <tvm/relay/base.h>
 #include <string>
 
 namespace tvm {

diff --git a/include/tvm/relay/attrs/image.h b/include/tvm/relay/attrs/image.h
@@ -25,6 +25,7 @@
 #define TVM_RELAY_ATTRS_IMAGE_H_
 
 #include <tvm/attrs.h>
+#include <tvm/relay/base.h>
 #include <string>
 
 namespace tvm {

diff --git a/include/tvm/relay/attrs/nn.h b/include/tvm/relay/attrs/nn.h
@@ -25,6 +25,7 @@
 #define TVM_RELAY_ATTRS_NN_H_
 
 #include <tvm/attrs.h>
+#include <tvm/relay/base.h>
 #include <string>
 
 namespace tvm {

diff --git a/include/tvm/relay/attrs/transform.h b/include/tvm/relay/attrs/transform.h
@@ -25,6 +25,7 @@
 #define TVM_RELAY_ATTRS_TRANSFORM_H_
 
 #include <tvm/attrs.h>
+#include <tvm/relay/base.h>
 #include <string>
 
 namespace tvm {

diff --git a/include/tvm/relay/attrs/vision.h b/include/tvm/relay/attrs/vision.h
@@ -25,6 +25,7 @@
 #define TVM_RELAY_ATTRS_VISION_H_
 
 #include <tvm/attrs.h>
+#include <tvm/relay/base.h>
 #include <string>
 
 namespace tvm {