[CPP] Refactor remove tvm/tvm.h (#3523)

include/tvm/tvm.h

-/*
- * Licensed to the Apache Software Foundation (ASF) under one
- * or more contributor license agreements.  See the NOTICE file
- * distributed with this work for additional information
- * regarding copyright ownership.  The ASF licenses this file
- * to you under the Apache License, Version 2.0 (the
- * "License"); you may not use this file except in compliance
- * with the License.  You may obtain a copy of the License at
- *
- *   http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing,
- * software distributed under the License is distributed on an
- * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
- * KIND, either express or implied.  See the License for the
- * specific language governing permissions and limitations
- * under the License.
- */
-
-/*!
- * \file tvm/tvm.h
- * \brief Header to include all C++ API.
- */
-#ifndef TVM_TVM_H_
-#define TVM_TVM_H_
-
-#include "base.h"
-#include "expr.h"
-#include "expr_operator.h"
-#include "tensor.h"
-#include "operation.h"
-#include "packed_func_ext.h"
-
-#endif  // TVM_TVM_H_

nnvm/src/compiler/alter_op_layout.cc

@@ -28,7 +28,7 @@
 #include <nnvm/layout.h>
 #include <nnvm/compiler/op_attr_types.h>
 #include <nnvm/pass_functions.h>
-#include <tvm/tvm.h>
+#include <tvm/operation.h>
 #include <algorithm>
 #include <functional>
 #include "compile_engine.h"

nnvm/src/top/image/resize.cc

@@ -22,7 +22,7 @@
  * \file resize.cc
  * \brief Property def of resize operators.
  */
-#include <tvm/tvm.h>
+#include <tvm/operation.h>
 #include <tvm/expr.h>
 #include <tvm/packed_func_ext.h>
 #include <nnvm/layout.h>

nnvm/src/top/nn/convolution.cc

@@ -30,7 +30,7 @@
 #include <tvm/tensor.h>
 #include <tvm/packed_func_ext.h>
 #include <nnvm/compiler/op_attr_types.h>
-#include <tvm/tvm.h>
+#include <tvm/operation.h>
 #include "nn_common.h"
 #include "../op_common.h"
 #include "../elemwise_op_common.h"

nnvm/src/top/nn/nn.cc

@@ -22,7 +22,7 @@
  * \file nn.cc
  * \brief Property def of nn operators.
  */
-#include <tvm/tvm.h>
+#include <tvm/operation.h>
 #include <tvm/expr.h>
 #include <tvm/packed_func_ext.h>
 #include <nnvm/op.h>

nnvm/src/top/nn/upsampling.cc

@@ -18,11 +18,10 @@
  */
 
 /*!
- *  Copyright (c) 2017 by Contributors
  * \file upsampling.cc
  * \brief Property def of upsampling operators.
  */
-#include <tvm/tvm.h>
+#include <tvm/operation.h>
 #include <tvm/expr.h>
 #include <nnvm/layout.h>
 #include <nnvm/compiler/op_attr_types.h>

src/relay/backend/utils.h

@@ -28,7 +28,6 @@
 #include <dmlc/json.h>
 #include <tvm/relay/expr.h>
 #include <tvm/relay/type.h>
-#include <tvm/tvm.h>
 #include <tvm/build_module.h>
 #include <tvm/codegen.h>
 #include <tvm/ir_pass.h>

src/relay/pass/alter_op_layout.cc

@@ -29,7 +29,7 @@
 #include <tvm/relay/op_attr_types.h>
 #include <tvm/relay/attrs/transform.h>
 #include <tvm/relay/transform.h>
-#include <tvm/tvm.h>
+#include <tvm/operation.h>
 #include <tuple>
 #include <vector>
 #include <functional>

tests/cpp/build_module_test.cc

@@ -20,8 +20,8 @@
 #include <dmlc/logging.h>
 #include <gtest/gtest.h>
 #include <topi/cuda/injective.h>
-#include <tvm/tvm.h>
 #include <tvm/operation.h>
+#include <tvm/packed_func_ext.h>
 #include <tvm/build_module.h>
 
 #include <string>

tests/cpp/container_test.cc

@@ -19,7 +19,7 @@
 
 #include <dmlc/logging.h>
 #include <gtest/gtest.h>
-#include <tvm/tvm.h>
+#include <tvm/packed_func_ext.h>
 
 TEST(Array, Expr) {
   using namespace tvm;

tests/cpp/expr_test.cc

@@ -19,7 +19,7 @@
 
 #include <dmlc/logging.h>
 #include <gtest/gtest.h>
-#include <tvm/tvm.h>
+#include <tvm/operation.h>
 
 TEST(Expr, Basic) {
   using namespace tvm;

tests/cpp/ir_functor_test.cc

@@ -19,7 +19,8 @@
 
 #include <dmlc/logging.h>
 #include <gtest/gtest.h>
-#include <tvm/tvm.h>
+#include <tvm/ir.h>
+#include <tvm/expr_operator.h>
 #include <tvm/node/ir_functor.h>
 #include <tvm/ir_functor_ext.h>
 

tests/cpp/ir_simplify_test.cc

@@ -20,7 +20,7 @@
 #include <dmlc/logging.h>
 #include <gtest/gtest.h>
 #include <tvm/ir_pass.h>
-#include <tvm/tvm.h>
+#include <tvm/operation.h>
 
 TEST(IRSIMPLIFY, MinMax) {
   auto x = tvm::var("x");

tests/cpp/ir_ssa_test.cc

@@ -19,7 +19,6 @@
 
 #include <dmlc/logging.h>
 #include <gtest/gtest.h>
-#include <tvm/tvm.h>
 #include <tvm/ir_pass.h>
 
 

tests/cpp/ir_visitor_test.cc

@@ -19,7 +19,6 @@
 
 #include <dmlc/logging.h>
 #include <gtest/gtest.h>
-#include <tvm/tvm.h>
 #include <tvm/ir_visitor.h>
 #include <tvm/ir_pass.h>
 

tests/cpp/packed_func_test.cc

@@ -20,7 +20,7 @@
 #include <dmlc/logging.h>
 #include <gtest/gtest.h>
 #include <tvm/runtime/packed_func.h>
-#include <tvm/tvm.h>
+#include <tvm/packed_func_ext.h>
 #include <tvm/ir.h>
 
 TEST(PackedFunc, Basic) {

tests/cpp/relay_build_module_test.cc

@@ -19,7 +19,7 @@
 
 #include <gtest/gtest.h>
 #include <tvm/build_module.h>
-#include <tvm/tvm.h>
+#include <tvm/operation.h>
 #include <tvm/relay/expr.h>
 #include <tvm/relay/type.h>
 #include <tvm/relay/analysis.h>

tests/cpp/relay_pass_type_infer_test.cc

@@ -18,7 +18,7 @@
  */
 
 #include <gtest/gtest.h>
-#include <tvm/tvm.h>
+#include <tvm/operation.h>
 #include <tvm/relay/expr.h>
 #include <tvm/relay/type.h>
 #include <tvm/relay/analysis.h>

tests/cpp/relay_transform_sequential.cc

@@ -28,7 +28,7 @@
 #include <tvm/relay/type.h>
 #include <tvm/runtime/packed_func.h>
 #include <tvm/runtime/registry.h>
-#include <tvm/tvm.h>
+#include <tvm/operation.h>
 
 TVM_REGISTER_GLOBAL("schedule")
     .set_body([](tvm::TVMArgs args, tvm::TVMRetValue* rv) {

tests/cpp/simple_passes_test.cc

@@ -19,8 +19,8 @@
 
 #include <dmlc/logging.h>
 #include <gtest/gtest.h>
-#include <tvm/tvm.h>
 #include <tvm/ir_pass.h>
+#include <tvm/operation.h>
 
 TEST(SimplePasses, HasSideEffect) {
   using namespace tvm;

tests/cpp/tensor_test.cc

@@ -19,7 +19,7 @@
 
 #include <dmlc/logging.h>
 #include <gtest/gtest.h>
-#include <tvm/tvm.h>
+#include <tvm/operation.h>
 
 TEST(Tensor, Basic) {
   using namespace tvm;

tests/cpp/topi_ewise_test.cc

@@ -17,7 +17,7 @@
  * under the License.
  */
 
-#include <tvm/tvm.h>
+#include <tvm/operation.h>
 #include <topi/elemwise.h>
 #include <gtest/gtest.h>
 

topi/include/topi/contrib/cublas.h

@@ -18,14 +18,13 @@
  */
 
 /*!
- *  Copyright (c) 2017 by Contributors
  * \brief External function interface to cuBLAS libraries
- * \file tags.h
+ * \file cublas.h
  */
 #ifndef TOPI_CONTRIB_CUBLAS_H_
 #define TOPI_CONTRIB_CUBLAS_H_
 
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "topi/detail/extern.h"
 
 namespace topi {

topi/include/topi/contrib/rocblas.h

@@ -25,7 +25,7 @@
 #ifndef TOPI_CONTRIB_ROCBLAS_H_
 #define TOPI_CONTRIB_ROCBLAS_H_
 
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "topi/detail/extern.h"
 
 namespace topi {

topi/include/topi/cuda/dense.h

@@ -18,14 +18,13 @@
  */
 
 /*!
-*  Copyright (c) 2017 by Contributors
-* \file cuda/dense.h
-* \brief CUDA schedule for dense operation
-*/
+ * \file cuda/dense.h
+ * \brief CUDA schedule for dense operation
+ */
 #ifndef TOPI_CUDA_DENSE_H_
 #define TOPI_CUDA_DENSE_H_
 
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/build_module.h"
 #include "topi/tags.h"
 #include "topi/detail/array_utils.h"
@@ -152,4 +151,3 @@ inline Schedule schedule_dense(const Target &target, const Array<Tensor>& outs)
 }  // namespace cuda
 }  // namespace topi
 #endif  // TOPI_CUDA_DENSE_H_
-

topi/include/topi/cuda/extern.h

@@ -18,16 +18,15 @@
  */
 
 /*!
-*  Copyright (c) 2017 by Contributors
-* \file cuda/extern.h
-* \brief CUDA schedule for extern followed by injective operations
-*/
+ * \file cuda/extern.h
+ * \brief CUDA schedule for extern followed by injective operations
+ */
 #ifndef TOPI_CUDA_EXTERN_H_
 #define TOPI_CUDA_EXTERN_H_
 
 #include "topi/tags.h"
 #include "topi/detail/fuse.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/build_module.h"
 
 namespace topi {

topi/include/topi/cuda/injective.h

@@ -18,16 +18,15 @@
  */
 
 /*!
-*  Copyright (c) 2017 by Contributors
-* \file cuda/injective.h
-* \brief CUDA schedule for injective operations
-*/
+ * \file cuda/injective.h
+ * \brief CUDA schedule for injective operations
+ */
 #ifndef TOPI_CUDA_INJECTIVE_H_
 #define TOPI_CUDA_INJECTIVE_H_
 
 #include "topi/tags.h"
 #include "topi/detail/fuse.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/build_module.h"
 
 namespace topi {

topi/include/topi/cuda/normalization.h

@@ -18,14 +18,13 @@
  */
 
 /*!
-*  Copyright (c) 2018 by Contributors
-* \file cuda/normalization.h
-* \brief CUDA schedule for LRN and l2 normalization operations
-*/
+ * \file cuda/normalization.h
+ * \brief CUDA schedule for LRN and l2 normalization operations
+ */
 #ifndef TOPI_CUDA_NORMALIZATION_H_
 #define TOPI_CUDA_NORMALIZATION_H_
 
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/build_module.h"
 #include "topi/tags.h"
 
@@ -122,4 +121,3 @@ inline Schedule schedule_l2_normalize(const Target &target, const Array<Tensor>&
 }  // namespace cuda
 }  // namespace topi
 #endif  // TOPI_CUDA_NORMALIZATION_H_
-

topi/include/topi/cuda/pooling.h

@@ -18,17 +18,16 @@
  */
 
 /*!
-*  Copyright (c) 2017 by Contributors
-* \file cuda/pooling.h
-* \brief CUDA schedule for pooling operations
-*/
+ * \file cuda/pooling.h
+ * \brief CUDA schedule for pooling operations
+ */
 #ifndef TOPI_CUDA_POOLING_H_
 #define TOPI_CUDA_POOLING_H_
 
 #include "topi/tags.h"
 #include "topi/detail/fuse.h"
 #include "topi/detail/array_utils.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/build_module.h"
 
 namespace topi {

topi/include/topi/cuda/reduction.h

@@ -18,16 +18,15 @@
  */
 
 /*!
-*  Copyright (c) 2017 by Contributors
-* \file cuda/reduction.h
-* \brief CUDA schedule for reduction operations
-*/
+ * \file cuda/reduction.h
+ * \brief CUDA schedule for reduction operations
+ */
 #ifndef TOPI_CUDA_REDUCTION_H_
 #define TOPI_CUDA_REDUCTION_H_
 
 #include "topi/tags.h"
 #include "topi/detail/fuse.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/build_module.h"
 
 namespace topi {

topi/include/topi/cuda/softmax.h

@@ -18,16 +18,15 @@
  */
 
 /*!
-*  Copyright (c) 2017 by Contributors
-* \file cuda/injective.h
-* \brief CUDA schedule for injective operations
-*/
+ * \file cuda/injective.h
+ * \brief CUDA schedule for injective operations
+ */
 #ifndef TOPI_CUDA_SOFTMAX_H_
 #define TOPI_CUDA_SOFTMAX_H_
 
 #include "topi/tags.h"
 #include "topi/detail/fuse.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/build_module.h"
 
 namespace topi {

topi/include/topi/detail/array_utils.h

@@ -18,14 +18,13 @@
  */
 
 /*!
-*  Copyright (c) 2017 by Contributors
-* \file array_utils.h
-* \brief Utility functions for handling arrays
-*/
+ * \file array_utils.h
+ * \brief Utility functions for handling arrays
+ */
 #ifndef TOPI_DETAIL_ARRAY_UTILS_H_
 #define TOPI_DETAIL_ARRAY_UTILS_H_
 
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 
 namespace topi {
 namespace detail {

topi/include/topi/detail/broadcast.h

@@ -30,7 +30,8 @@
 #include <string>
 
 #include "tvm/ir_pass.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
+#include "tvm/expr_operator.h"
 #include "topi/detail/constant_utils.h"
 
 namespace topi {

topi/include/topi/detail/constant_utils.h

@@ -28,7 +28,7 @@
 #include <string>
 #include <vector>
 
-#include "tvm/tvm.h"
+#include "tvm/expr.h"
 #include "tvm/ir_pass.h"
 
 namespace topi {

topi/include/topi/detail/extern.h

@@ -18,14 +18,13 @@
  */
 
 /*!
-*  Copyright (c) 2017 by Contributors
-* \file detail/extern.h
-* \brief Helpers for using external functions
-*/
+ * \file detail/extern.h
+ * \brief Helpers for using external functions
+ */
 #ifndef TOPI_DETAIL_EXTERN_H_
 #define TOPI_DETAIL_EXTERN_H_
 
-#include <tvm/tvm.h>
+#include <tvm/operation.h>
 #include <vector>
 #include <string>
 

topi/include/topi/detail/fuse.h

@@ -18,14 +18,13 @@
  */
 
 /*!
-*  Copyright (c) 2017 by Contributors
-* \file fuse.h
-* \brief Fuse operation
-*/
+ * \file fuse.h
+ * \brief Fuse operation
+ */
 #ifndef TOPI_DETAIL_FUSE_H_
 #define TOPI_DETAIL_FUSE_H_
 
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 
 namespace topi {
 namespace detail {

topi/include/topi/detail/pad_utils.h

@@ -27,7 +27,8 @@
 
 #include <vector>
 
-#include "tvm/tvm.h"
+#include "tvm/expr.h"
+#include "tvm/expr_operator.h"
 
 namespace topi {
 namespace detail {

topi/include/topi/detail/ravel_unravel.h

@@ -27,7 +27,8 @@
 
 #include <vector>
 
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
+#include "tvm/expr_operator.h"
 
 namespace topi {
 namespace detail {

topi/include/topi/elemwise.h

@@ -28,7 +28,6 @@
 #include <string>
 
 #include "topi/tags.h"
-#include "tvm/tvm.h"
 #include "tvm/ir.h"
 #include "tvm/ir_pass.h"
 #include "broadcast.h"

topi/include/topi/generic/default.h

@@ -18,16 +18,15 @@
  */
 
 /*!
-*  Copyright (c) 2017 by Contributors
-* \file generic/default.h
-* \brief Generic default schedule
-*/
+ * \file generic/default.h
+ * \brief Generic default schedule
+ */
 #ifndef TOPI_GENERIC_DEFAULT_H_
 #define TOPI_GENERIC_DEFAULT_H_
 
 #include "topi/tags.h"
 #include "topi/detail/fuse.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/build_module.h"
 
 namespace topi {

topi/include/topi/generic/extern.h

@@ -18,16 +18,15 @@
  */
 
 /*!
-*  Copyright (c) 2017 by Contributors
-* \file generic/extern.h
-* \brief Schedule for extern followed by injective ops
-*/
+ * \file generic/extern.h
+ * \brief Schedule for extern followed by injective ops
+ */
 #ifndef TOPI_GENERIC_EXTERN_H_
 #define TOPI_GENERIC_EXTERN_H_
 
 #include "topi/tags.h"
 #include "topi/detail/fuse.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/build_module.h"
 
 namespace topi {

topi/include/topi/generic/injective.h

@@ -18,16 +18,15 @@
  */
 
 /*!
-*  Copyright (c) 2017 by Contributors
-* \file generic/injective.h
-* \brief Generic schedule for injective operations
-*/
+ * \file generic/injective.h
+ * \brief Generic schedule for injective operations
+ */
 #ifndef TOPI_GENERIC_INJECTIVE_H_
 #define TOPI_GENERIC_INJECTIVE_H_
 
 #include "topi/tags.h"
 #include "topi/detail/fuse.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/build_module.h"
 
 namespace topi {

topi/include/topi/image/resize.h

@@ -34,7 +34,8 @@
 #include "topi/elemwise.h"
 #include "topi/detail/ravel_unravel.h"
 #include "topi/detail/constant_utils.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
+#include "tvm/expr_operator.h"
 
 namespace topi {
 namespace image {

topi/include/topi/nn.h

@@ -32,7 +32,8 @@
 #include "topi/detail/constant_utils.h"
 #include "tvm/ir.h"
 #include "tvm/ir_pass.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
+#include "tvm/expr_operator.h"
 
 namespace topi {
 using namespace tvm;

topi/include/topi/nn/batch_matmul.h

@@ -28,7 +28,7 @@
 #include <string>
 
 #include "topi/tags.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 
 namespace topi {
 namespace nn {

topi/include/topi/nn/bias_add.h

@@ -30,7 +30,8 @@
 #include "topi/tags.h"
 #include "topi/broadcast.h"
 #include "topi/transform.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
+#include "tvm/expr_operator.h"
 
 namespace topi {
 namespace nn {

topi/include/topi/nn/bnn.h

@@ -18,7 +18,6 @@
  */
 
 /*!
- *  Copyright (c) 2017 by Contributors
  * \brief Binary op constructions
  * \file nn/bnn.h
  */
@@ -27,7 +26,7 @@
 
 #include <string>
 
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/ir_pass.h"
 #include "topi/tags.h"
 #include "topi/detail/constant_utils.h"

topi/include/topi/nn/dense.h

@@ -28,7 +28,7 @@
 #include <string>
 
 #include "topi/tags.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 
 namespace topi {
 namespace nn {

topi/include/topi/nn/dilate.h

@@ -27,7 +27,7 @@
 
 #include <string>
 
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/ir_pass.h"
 #include "topi/tags.h"
 

topi/include/topi/nn/flatten.h

@@ -30,7 +30,9 @@
 
 #include "topi/tags.h"
 #include "topi/detail/constant_utils.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
+#include "tvm/expr_operator.h"
+
 
 namespace topi {
 namespace nn {

topi/include/topi/nn/l2_normalize.h

@@ -18,7 +18,6 @@
  */
 
 /*!
- *  Copyright (c) 2018 by Contributors
  * \brief l2 normalization op constructions
  * \file nn/l2_normalize.h
  */
@@ -28,7 +27,7 @@
 #include <string>
 #include <algorithm>
 #include "topi/tags.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 namespace topi {
 namespace nn {
 using namespace tvm;

topi/include/topi/nn/local_response_norm.h

@@ -18,7 +18,6 @@
  */
 
 /*!
- *  Copyright (c) 2018 by Contributors
  * \brief local response normalization op constructions
  * \file nn/local_response_norm.h
  */
@@ -28,7 +27,7 @@
 #include <string>
 
 #include "topi/tags.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 
 namespace topi {
 namespace nn {

topi/include/topi/nn/mapping.h

@@ -18,7 +18,6 @@
  */
 
 /*!
- *  Copyright (c) 2017 by Contributors
  * \brief Mapping op constructions
  * \file nn/mapping.h
  */
@@ -28,7 +27,7 @@
 #include <string>
 
 #include "topi/tags.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 
 namespace topi {
 namespace nn {

topi/include/topi/nn/pooling.h

@@ -28,7 +28,6 @@
 #include <string>
 #include <vector>
 
-#include "tvm/tvm.h"
 #include "tvm/ir_pass.h"
 #include "topi/tags.h"
 #include "topi/detail/pad_utils.h"

topi/include/topi/nn/softmax.h

@@ -30,7 +30,8 @@
 
 #include "topi/reduction.h"
 #include "topi/tags.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
+#include "tvm/expr_operator.h"
 
 namespace topi {
 namespace nn {

topi/include/topi/reduction.h

@@ -36,7 +36,8 @@
 #include "topi/transform.h"
 #include "topi/detail/ravel_unravel.h"
 #include "topi/detail/constant_utils.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
+#include "tvm/expr_operator.h"
 
 
 namespace topi {

topi/include/topi/rocm/dense.h

@@ -18,14 +18,13 @@
  */
 
 /*!
-*  Copyright (c) 2017 by Contributors
-* \file rocm/dense.h
-* \brief rocm schedule for dense operation
-*/
+ * \file rocm/dense.h
+ * \brief rocm schedule for dense operation
+ */
 #ifndef TOPI_ROCM_DENSE_H_
 #define TOPI_ROCM_DENSE_H_
 
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/build_module.h"
 #include "topi/tags.h"
 #include "topi/detail/array_utils.h"
@@ -100,4 +99,3 @@ inline Schedule schedule_dense(const Target &target, const Array<Tensor>& outs)
 }  // namespace rocm
 }  // namespace topi
 #endif  // TOPI_ROCM_DENSE_H_
-

topi/include/topi/rocm/normalization.h

@@ -18,14 +18,13 @@
  */
 
 /*!
-*  Copyright (c) 2018 by Contributors
-* \file rocm/normalization.h
-* \brief rocm schedule for LRN and l2 normalization operations
-*/
+ * \file rocm/normalization.h
+ * \brief rocm schedule for LRN and l2 normalization operations
+ */
 #ifndef TOPI_ROCM_NORMALIZATION_H_
 #define TOPI_ROCM_NORMALIZATION_H_
 
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/build_module.h"
 #include "topi/tags.h"
 

topi/include/topi/transform.h

@@ -35,7 +35,8 @@
 #include "topi/tags.h"
 #include "topi/detail/ravel_unravel.h"
 #include "topi/detail/constant_utils.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
+#include "tvm/expr_operator.h"
 #include "tvm/data_layout.h"
 
 namespace topi {

topi/include/topi/vision/reorg.h

@@ -32,7 +32,8 @@
 #include "topi/reduction.h"
 #include "topi/tags.h"
 #include "topi/transform.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
+#include "tvm/expr_operator.h"
 
 namespace topi {
 namespace vision {

topi/include/topi/x86/bnn.h

@@ -18,16 +18,15 @@
  */
 
 /*!
-*  Copyright (c) 2017 by Contributors
-* \file x86/bnn.h
-* \brief x86 schedule for binary operations
-*/
+ * \file x86/bnn.h
+ * \brief x86 schedule for binary operations
+ */
 #ifndef TOPI_X86_BNN_H_
 #define TOPI_X86_BNN_H_
 
 #include "topi/tags.h"
 #include "topi/detail/fuse.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/build_module.h"
 
 namespace topi {

topi/include/topi/x86/default.h

@@ -18,16 +18,15 @@
  */
 
 /*!
-*  Copyright (c) 2017 by Contributors
-* \file x86/default.h
-* \brief default x86 schedule
-*/
+ * \file x86/default.h
+ * \brief default x86 schedule
+ */
 #ifndef TOPI_X86_DEFAULT_H_
 #define TOPI_X86_DEFAULT_H_
 
 #include "topi/tags.h"
 #include "topi/detail/fuse.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/build_module.h"
 
 namespace topi {

topi/include/topi/x86/injective.h

@@ -18,16 +18,15 @@
  */
 
 /*!
-*  Copyright (c) 2017 by Contributors
-* \file x86/injective.h
-* \brief x86 schedule for injective ops
-*/
+ * \file x86/injective.h
+ * \brief x86 schedule for injective ops
+ */
 #ifndef TOPI_X86_INJECTIVE_H_
 #define TOPI_X86_INJECTIVE_H_
 
 #include "topi/tags.h"
 #include "topi/detail/fuse.h"
-#include "tvm/tvm.h"
+#include "tvm/operation.h"
 #include "tvm/build_module.h"
 
 namespace topi {