From 8c278154cc98643a2845891e54ab3c4940522c28 Mon Sep 17 00:00:00 2001 From: hlu1 <14827759+hlu1@users.noreply.github.com> Date: Tue, 18 Jun 2019 15:26:27 -0700 Subject: [PATCH] [Bugfix] Missing headers --- include/tvm/relay/attrs/algorithm.h | 1 + include/tvm/relay/attrs/image.h | 1 + include/tvm/relay/attrs/nn.h | 1 + include/tvm/relay/attrs/transform.h | 1 + include/tvm/relay/attrs/vision.h | 1 + 5 files changed, 5 insertions(+) diff --git a/include/tvm/relay/attrs/algorithm.h b/include/tvm/relay/attrs/algorithm.h index f5ba6999347f..ce14a6a2d535 100644 --- 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 +#include #include namespace tvm { diff --git a/include/tvm/relay/attrs/image.h b/include/tvm/relay/attrs/image.h index f3caf213575d..e74c0f53ad65 100644 --- 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 +#include #include namespace tvm { diff --git a/include/tvm/relay/attrs/nn.h b/include/tvm/relay/attrs/nn.h index 8a1aca0a4b4a..0fbf984817b9 100644 --- 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 +#include #include namespace tvm { diff --git a/include/tvm/relay/attrs/transform.h b/include/tvm/relay/attrs/transform.h index 65febae5e7ff..5e315185ea7f 100644 --- 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 +#include #include namespace tvm { diff --git a/include/tvm/relay/attrs/vision.h b/include/tvm/relay/attrs/vision.h index 7fa1ffb8a4fe..b98bbfc5988e 100644 --- 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 +#include #include namespace tvm {