Skip to content
Projects
Groups
Snippets
Help
This project
Loading...
Sign in / Register
Toggle navigation
T
tic
Overview
Overview
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
wenyuanbo
tic
Commits
025a6c80
Commit
025a6c80
authored
Jul 09, 2019
by
Tianqi Chen
Committed by
Jared Roesch
Jul 09, 2019
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
[CPP] Refactor remove tvm/tvm.h (#3523)
parent
db841c24
Hide whitespace changes
Inline
Side-by-side
Showing
63 changed files
with
233 additions
and
286 deletions
+233
-286
include/tvm/tvm.h
+0
-34
nnvm/src/compiler/alter_op_layout.cc
+3
-3
nnvm/src/top/image/resize.cc
+3
-3
nnvm/src/top/nn/convolution.cc
+3
-3
nnvm/src/top/nn/nn.cc
+3
-3
nnvm/src/top/nn/upsampling.cc
+3
-4
src/relay/backend/utils.h
+2
-3
src/relay/pass/alter_op_layout.cc
+1
-1
tests/cpp/build_module_test.cc
+1
-1
tests/cpp/container_test.cc
+3
-3
tests/cpp/expr_test.cc
+3
-3
tests/cpp/ir_functor_test.cc
+4
-3
tests/cpp/ir_simplify_test.cc
+1
-1
tests/cpp/ir_ssa_test.cc
+2
-3
tests/cpp/ir_visitor_test.cc
+2
-3
tests/cpp/packed_func_test.cc
+3
-3
tests/cpp/relay_build_module_test.cc
+1
-1
tests/cpp/relay_pass_type_infer_test.cc
+3
-3
tests/cpp/relay_transform_sequential.cc
+1
-1
tests/cpp/simple_passes_test.cc
+3
-3
tests/cpp/tensor_test.cc
+3
-3
tests/cpp/topi_ewise_test.cc
+3
-3
topi/include/topi/contrib/cublas.h
+4
-5
topi/include/topi/contrib/rocblas.h
+3
-3
topi/include/topi/cuda/dense.h
+6
-8
topi/include/topi/cuda/extern.h
+6
-7
topi/include/topi/cuda/injective.h
+6
-7
topi/include/topi/cuda/normalization.h
+6
-8
topi/include/topi/cuda/pooling.h
+6
-7
topi/include/topi/cuda/reduction.h
+6
-7
topi/include/topi/cuda/softmax.h
+6
-7
topi/include/topi/detail/array_utils.h
+6
-7
topi/include/topi/detail/broadcast.h
+4
-3
topi/include/topi/detail/constant_utils.h
+3
-3
topi/include/topi/detail/extern.h
+6
-7
topi/include/topi/detail/fuse.h
+6
-7
topi/include/topi/detail/pad_utils.h
+4
-3
topi/include/topi/detail/ravel_unravel.h
+4
-3
topi/include/topi/elemwise.h
+0
-1
topi/include/topi/generic/default.h
+6
-7
topi/include/topi/generic/extern.h
+6
-7
topi/include/topi/generic/injective.h
+6
-7
topi/include/topi/image/resize.h
+4
-3
topi/include/topi/nn.h
+4
-3
topi/include/topi/nn/batch_matmul.h
+3
-3
topi/include/topi/nn/bias_add.h
+2
-1
topi/include/topi/nn/bnn.h
+3
-4
topi/include/topi/nn/dense.h
+3
-3
topi/include/topi/nn/dilate.h
+3
-3
topi/include/topi/nn/flatten.h
+5
-3
topi/include/topi/nn/l2_normalize.h
+3
-4
topi/include/topi/nn/local_response_norm.h
+3
-4
topi/include/topi/nn/mapping.h
+3
-4
topi/include/topi/nn/pooling.h
+2
-3
topi/include/topi/nn/softmax.h
+4
-3
topi/include/topi/reduction.h
+4
-3
topi/include/topi/rocm/dense.h
+6
-8
topi/include/topi/rocm/normalization.h
+6
-7
topi/include/topi/transform.h
+2
-1
topi/include/topi/vision/reorg.h
+4
-3
topi/include/topi/x86/bnn.h
+6
-7
topi/include/topi/x86/default.h
+6
-7
topi/include/topi/x86/injective.h
+6
-7
No files found.
include/tvm/tvm.h
deleted
100644 → 0
View file @
db841c24
/*
* 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
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -28,7 +28,7 @@
...
@@ -28,7 +28,7 @@
#include <nnvm/layout.h>
#include <nnvm/layout.h>
#include <nnvm/compiler/op_attr_types.h>
#include <nnvm/compiler/op_attr_types.h>
#include <nnvm/pass_functions.h>
#include <nnvm/pass_functions.h>
#include <tvm/
tvm
.h>
#include <tvm/
operation
.h>
#include <algorithm>
#include <algorithm>
#include <functional>
#include <functional>
#include "compile_engine.h"
#include "compile_engine.h"
...
...
nnvm/src/top/image/resize.cc
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -22,7 +22,7 @@
...
@@ -22,7 +22,7 @@
* \file resize.cc
* \file resize.cc
* \brief Property def of resize operators.
* \brief Property def of resize operators.
*/
*/
#include <tvm/
tvm
.h>
#include <tvm/
operation
.h>
#include <tvm/expr.h>
#include <tvm/expr.h>
#include <tvm/packed_func_ext.h>
#include <tvm/packed_func_ext.h>
#include <nnvm/layout.h>
#include <nnvm/layout.h>
...
...
nnvm/src/top/nn/convolution.cc
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -30,7 +30,7 @@
...
@@ -30,7 +30,7 @@
#include <tvm/tensor.h>
#include <tvm/tensor.h>
#include <tvm/packed_func_ext.h>
#include <tvm/packed_func_ext.h>
#include <nnvm/compiler/op_attr_types.h>
#include <nnvm/compiler/op_attr_types.h>
#include <tvm/
tvm
.h>
#include <tvm/
operation
.h>
#include "nn_common.h"
#include "nn_common.h"
#include "../op_common.h"
#include "../op_common.h"
#include "../elemwise_op_common.h"
#include "../elemwise_op_common.h"
...
...
nnvm/src/top/nn/nn.cc
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -22,7 +22,7 @@
...
@@ -22,7 +22,7 @@
* \file nn.cc
* \file nn.cc
* \brief Property def of nn operators.
* \brief Property def of nn operators.
*/
*/
#include <tvm/
tvm
.h>
#include <tvm/
operation
.h>
#include <tvm/expr.h>
#include <tvm/expr.h>
#include <tvm/packed_func_ext.h>
#include <tvm/packed_func_ext.h>
#include <nnvm/op.h>
#include <nnvm/op.h>
...
...
nnvm/src/top/nn/upsampling.cc
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,11 +18,10 @@
...
@@ -18,11 +18,10 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file upsampling.cc
* \file upsampling.cc
* \brief Property def of upsampling operators.
* \brief Property def of upsampling operators.
*/
*/
#include <tvm/
tvm
.h>
#include <tvm/
operation
.h>
#include <tvm/expr.h>
#include <tvm/expr.h>
#include <nnvm/layout.h>
#include <nnvm/layout.h>
#include <nnvm/compiler/op_attr_types.h>
#include <nnvm/compiler/op_attr_types.h>
...
...
src/relay/backend/utils.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -28,7 +28,6 @@
...
@@ -28,7 +28,6 @@
#include <dmlc/json.h>
#include <dmlc/json.h>
#include <tvm/relay/expr.h>
#include <tvm/relay/expr.h>
#include <tvm/relay/type.h>
#include <tvm/relay/type.h>
#include <tvm/tvm.h>
#include <tvm/build_module.h>
#include <tvm/build_module.h>
#include <tvm/codegen.h>
#include <tvm/codegen.h>
#include <tvm/ir_pass.h>
#include <tvm/ir_pass.h>
...
...
src/relay/pass/alter_op_layout.cc
View file @
025a6c80
...
@@ -29,7 +29,7 @@
...
@@ -29,7 +29,7 @@
#include <tvm/relay/op_attr_types.h>
#include <tvm/relay/op_attr_types.h>
#include <tvm/relay/attrs/transform.h>
#include <tvm/relay/attrs/transform.h>
#include <tvm/relay/transform.h>
#include <tvm/relay/transform.h>
#include <tvm/
tvm
.h>
#include <tvm/
operation
.h>
#include <tuple>
#include <tuple>
#include <vector>
#include <vector>
#include <functional>
#include <functional>
...
...
tests/cpp/build_module_test.cc
View file @
025a6c80
...
@@ -20,8 +20,8 @@
...
@@ -20,8 +20,8 @@
#include <dmlc/logging.h>
#include <dmlc/logging.h>
#include <gtest/gtest.h>
#include <gtest/gtest.h>
#include <topi/cuda/injective.h>
#include <topi/cuda/injective.h>
#include <tvm/tvm.h>
#include <tvm/operation.h>
#include <tvm/operation.h>
#include <tvm/packed_func_ext.h>
#include <tvm/build_module.h>
#include <tvm/build_module.h>
#include <string>
#include <string>
...
...
tests/cpp/container_test.cc
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -19,7 +19,7 @@
...
@@ -19,7 +19,7 @@
#include <dmlc/logging.h>
#include <dmlc/logging.h>
#include <gtest/gtest.h>
#include <gtest/gtest.h>
#include <tvm/
tvm
.h>
#include <tvm/
packed_func_ext
.h>
TEST
(
Array
,
Expr
)
{
TEST
(
Array
,
Expr
)
{
using
namespace
tvm
;
using
namespace
tvm
;
...
...
tests/cpp/expr_test.cc
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -19,7 +19,7 @@
...
@@ -19,7 +19,7 @@
#include <dmlc/logging.h>
#include <dmlc/logging.h>
#include <gtest/gtest.h>
#include <gtest/gtest.h>
#include <tvm/
tvm
.h>
#include <tvm/
operation
.h>
TEST
(
Expr
,
Basic
)
{
TEST
(
Expr
,
Basic
)
{
using
namespace
tvm
;
using
namespace
tvm
;
...
...
tests/cpp/ir_functor_test.cc
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -19,7 +19,8 @@
...
@@ -19,7 +19,8 @@
#include <dmlc/logging.h>
#include <dmlc/logging.h>
#include <gtest/gtest.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/node/ir_functor.h>
#include <tvm/ir_functor_ext.h>
#include <tvm/ir_functor_ext.h>
...
...
tests/cpp/ir_simplify_test.cc
View file @
025a6c80
...
@@ -20,7 +20,7 @@
...
@@ -20,7 +20,7 @@
#include <dmlc/logging.h>
#include <dmlc/logging.h>
#include <gtest/gtest.h>
#include <gtest/gtest.h>
#include <tvm/ir_pass.h>
#include <tvm/ir_pass.h>
#include <tvm/
tvm
.h>
#include <tvm/
operation
.h>
TEST
(
IRSIMPLIFY
,
MinMax
)
{
TEST
(
IRSIMPLIFY
,
MinMax
)
{
auto
x
=
tvm
::
var
(
"x"
);
auto
x
=
tvm
::
var
(
"x"
);
...
...
tests/cpp/ir_ssa_test.cc
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -19,7 +19,6 @@
...
@@ -19,7 +19,6 @@
#include <dmlc/logging.h>
#include <dmlc/logging.h>
#include <gtest/gtest.h>
#include <gtest/gtest.h>
#include <tvm/tvm.h>
#include <tvm/ir_pass.h>
#include <tvm/ir_pass.h>
...
...
tests/cpp/ir_visitor_test.cc
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -19,7 +19,6 @@
...
@@ -19,7 +19,6 @@
#include <dmlc/logging.h>
#include <dmlc/logging.h>
#include <gtest/gtest.h>
#include <gtest/gtest.h>
#include <tvm/tvm.h>
#include <tvm/ir_visitor.h>
#include <tvm/ir_visitor.h>
#include <tvm/ir_pass.h>
#include <tvm/ir_pass.h>
...
...
tests/cpp/packed_func_test.cc
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -20,7 +20,7 @@
...
@@ -20,7 +20,7 @@
#include <dmlc/logging.h>
#include <dmlc/logging.h>
#include <gtest/gtest.h>
#include <gtest/gtest.h>
#include <tvm/runtime/packed_func.h>
#include <tvm/runtime/packed_func.h>
#include <tvm/
tvm
.h>
#include <tvm/
packed_func_ext
.h>
#include <tvm/ir.h>
#include <tvm/ir.h>
TEST
(
PackedFunc
,
Basic
)
{
TEST
(
PackedFunc
,
Basic
)
{
...
...
tests/cpp/relay_build_module_test.cc
View file @
025a6c80
...
@@ -19,7 +19,7 @@
...
@@ -19,7 +19,7 @@
#include <gtest/gtest.h>
#include <gtest/gtest.h>
#include <tvm/build_module.h>
#include <tvm/build_module.h>
#include <tvm/
tvm
.h>
#include <tvm/
operation
.h>
#include <tvm/relay/expr.h>
#include <tvm/relay/expr.h>
#include <tvm/relay/type.h>
#include <tvm/relay/type.h>
#include <tvm/relay/analysis.h>
#include <tvm/relay/analysis.h>
...
...
tests/cpp/relay_pass_type_infer_test.cc
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,7 +18,7 @@
...
@@ -18,7 +18,7 @@
*/
*/
#include <gtest/gtest.h>
#include <gtest/gtest.h>
#include <tvm/
tvm
.h>
#include <tvm/
operation
.h>
#include <tvm/relay/expr.h>
#include <tvm/relay/expr.h>
#include <tvm/relay/type.h>
#include <tvm/relay/type.h>
#include <tvm/relay/analysis.h>
#include <tvm/relay/analysis.h>
...
...
tests/cpp/relay_transform_sequential.cc
View file @
025a6c80
...
@@ -28,7 +28,7 @@
...
@@ -28,7 +28,7 @@
#include <tvm/relay/type.h>
#include <tvm/relay/type.h>
#include <tvm/runtime/packed_func.h>
#include <tvm/runtime/packed_func.h>
#include <tvm/runtime/registry.h>
#include <tvm/runtime/registry.h>
#include <tvm/
tvm
.h>
#include <tvm/
operation
.h>
TVM_REGISTER_GLOBAL
(
"schedule"
)
TVM_REGISTER_GLOBAL
(
"schedule"
)
.
set_body
([](
tvm
::
TVMArgs
args
,
tvm
::
TVMRetValue
*
rv
)
{
.
set_body
([](
tvm
::
TVMArgs
args
,
tvm
::
TVMRetValue
*
rv
)
{
...
...
tests/cpp/simple_passes_test.cc
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -19,8 +19,8 @@
...
@@ -19,8 +19,8 @@
#include <dmlc/logging.h>
#include <dmlc/logging.h>
#include <gtest/gtest.h>
#include <gtest/gtest.h>
#include <tvm/tvm.h>
#include <tvm/ir_pass.h>
#include <tvm/ir_pass.h>
#include <tvm/operation.h>
TEST
(
SimplePasses
,
HasSideEffect
)
{
TEST
(
SimplePasses
,
HasSideEffect
)
{
using
namespace
tvm
;
using
namespace
tvm
;
...
...
tests/cpp/tensor_test.cc
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -19,7 +19,7 @@
...
@@ -19,7 +19,7 @@
#include <dmlc/logging.h>
#include <dmlc/logging.h>
#include <gtest/gtest.h>
#include <gtest/gtest.h>
#include <tvm/
tvm
.h>
#include <tvm/
operation
.h>
TEST
(
Tensor
,
Basic
)
{
TEST
(
Tensor
,
Basic
)
{
using
namespace
tvm
;
using
namespace
tvm
;
...
...
tests/cpp/topi_ewise_test.cc
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -17,7 +17,7 @@
...
@@ -17,7 +17,7 @@
* under the License.
* under the License.
*/
*/
#include <tvm/
tvm
.h>
#include <tvm/
operation
.h>
#include <topi/elemwise.h>
#include <topi/elemwise.h>
#include <gtest/gtest.h>
#include <gtest/gtest.h>
...
...
topi/include/topi/contrib/cublas.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,14 +18,13 @@
...
@@ -18,14 +18,13 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \brief External function interface to cuBLAS libraries
* \brief External function interface to cuBLAS libraries
* \file
tag
s.h
* \file
cubla
s.h
*/
*/
#ifndef TOPI_CONTRIB_CUBLAS_H_
#ifndef TOPI_CONTRIB_CUBLAS_H_
#define TOPI_CONTRIB_CUBLAS_H_
#define TOPI_CONTRIB_CUBLAS_H_
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "topi/detail/extern.h"
#include "topi/detail/extern.h"
namespace
topi
{
namespace
topi
{
...
...
topi/include/topi/contrib/rocblas.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -25,7 +25,7 @@
...
@@ -25,7 +25,7 @@
#ifndef TOPI_CONTRIB_ROCBLAS_H_
#ifndef TOPI_CONTRIB_ROCBLAS_H_
#define TOPI_CONTRIB_ROCBLAS_H_
#define TOPI_CONTRIB_ROCBLAS_H_
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "topi/detail/extern.h"
#include "topi/detail/extern.h"
namespace
topi
{
namespace
topi
{
...
...
topi/include/topi/cuda/dense.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,14 +18,13 @@
...
@@ -18,14 +18,13 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file cuda/dense.h
* \file cuda/dense.h
* \brief CUDA schedule for dense operation
* \brief CUDA schedule for dense operation
*/
*/
#ifndef TOPI_CUDA_DENSE_H_
#ifndef TOPI_CUDA_DENSE_H_
#define TOPI_CUDA_DENSE_H_
#define TOPI_CUDA_DENSE_H_
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/build_module.h"
#include "tvm/build_module.h"
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/array_utils.h"
#include "topi/detail/array_utils.h"
...
@@ -152,4 +151,3 @@ inline Schedule schedule_dense(const Target &target, const Array<Tensor>& outs)
...
@@ -152,4 +151,3 @@ inline Schedule schedule_dense(const Target &target, const Array<Tensor>& outs)
}
// namespace cuda
}
// namespace cuda
}
// namespace topi
}
// namespace topi
#endif // TOPI_CUDA_DENSE_H_
#endif // TOPI_CUDA_DENSE_H_
topi/include/topi/cuda/extern.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,16 +18,15 @@
...
@@ -18,16 +18,15 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file cuda/extern.h
* \file cuda/extern.h
* \brief CUDA schedule for extern followed by injective operations
* \brief CUDA schedule for extern followed by injective operations
*/
*/
#ifndef TOPI_CUDA_EXTERN_H_
#ifndef TOPI_CUDA_EXTERN_H_
#define TOPI_CUDA_EXTERN_H_
#define TOPI_CUDA_EXTERN_H_
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/fuse.h"
#include "topi/detail/fuse.h"
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/build_module.h"
#include "tvm/build_module.h"
namespace
topi
{
namespace
topi
{
...
...
topi/include/topi/cuda/injective.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,16 +18,15 @@
...
@@ -18,16 +18,15 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file cuda/injective.h
* \file cuda/injective.h
* \brief CUDA schedule for injective operations
* \brief CUDA schedule for injective operations
*/
*/
#ifndef TOPI_CUDA_INJECTIVE_H_
#ifndef TOPI_CUDA_INJECTIVE_H_
#define TOPI_CUDA_INJECTIVE_H_
#define TOPI_CUDA_INJECTIVE_H_
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/fuse.h"
#include "topi/detail/fuse.h"
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/build_module.h"
#include "tvm/build_module.h"
namespace
topi
{
namespace
topi
{
...
...
topi/include/topi/cuda/normalization.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,14 +18,13 @@
...
@@ -18,14 +18,13 @@
*/
*/
/*!
/*!
* Copyright (c) 2018 by Contributors
* \file cuda/normalization.h
* \file cuda/normalization.h
* \brief CUDA schedule for LRN and l2 normalization operations
* \brief CUDA schedule for LRN and l2 normalization operations
*/
*/
#ifndef TOPI_CUDA_NORMALIZATION_H_
#ifndef TOPI_CUDA_NORMALIZATION_H_
#define TOPI_CUDA_NORMALIZATION_H_
#define TOPI_CUDA_NORMALIZATION_H_
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/build_module.h"
#include "tvm/build_module.h"
#include "topi/tags.h"
#include "topi/tags.h"
...
@@ -122,4 +121,3 @@ inline Schedule schedule_l2_normalize(const Target &target, const Array<Tensor>&
...
@@ -122,4 +121,3 @@ inline Schedule schedule_l2_normalize(const Target &target, const Array<Tensor>&
}
// namespace cuda
}
// namespace cuda
}
// namespace topi
}
// namespace topi
#endif // TOPI_CUDA_NORMALIZATION_H_
#endif // TOPI_CUDA_NORMALIZATION_H_
topi/include/topi/cuda/pooling.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,17 +18,16 @@
...
@@ -18,17 +18,16 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file cuda/pooling.h
* \file cuda/pooling.h
* \brief CUDA schedule for pooling operations
* \brief CUDA schedule for pooling operations
*/
*/
#ifndef TOPI_CUDA_POOLING_H_
#ifndef TOPI_CUDA_POOLING_H_
#define TOPI_CUDA_POOLING_H_
#define TOPI_CUDA_POOLING_H_
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/fuse.h"
#include "topi/detail/fuse.h"
#include "topi/detail/array_utils.h"
#include "topi/detail/array_utils.h"
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/build_module.h"
#include "tvm/build_module.h"
namespace
topi
{
namespace
topi
{
...
...
topi/include/topi/cuda/reduction.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,16 +18,15 @@
...
@@ -18,16 +18,15 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file cuda/reduction.h
* \file cuda/reduction.h
* \brief CUDA schedule for reduction operations
* \brief CUDA schedule for reduction operations
*/
*/
#ifndef TOPI_CUDA_REDUCTION_H_
#ifndef TOPI_CUDA_REDUCTION_H_
#define TOPI_CUDA_REDUCTION_H_
#define TOPI_CUDA_REDUCTION_H_
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/fuse.h"
#include "topi/detail/fuse.h"
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/build_module.h"
#include "tvm/build_module.h"
namespace
topi
{
namespace
topi
{
...
...
topi/include/topi/cuda/softmax.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,16 +18,15 @@
...
@@ -18,16 +18,15 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file cuda/injective.h
* \file cuda/injective.h
* \brief CUDA schedule for injective operations
* \brief CUDA schedule for injective operations
*/
*/
#ifndef TOPI_CUDA_SOFTMAX_H_
#ifndef TOPI_CUDA_SOFTMAX_H_
#define TOPI_CUDA_SOFTMAX_H_
#define TOPI_CUDA_SOFTMAX_H_
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/fuse.h"
#include "topi/detail/fuse.h"
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/build_module.h"
#include "tvm/build_module.h"
namespace
topi
{
namespace
topi
{
...
...
topi/include/topi/detail/array_utils.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,14 +18,13 @@
...
@@ -18,14 +18,13 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file array_utils.h
* \file array_utils.h
* \brief Utility functions for handling arrays
* \brief Utility functions for handling arrays
*/
*/
#ifndef TOPI_DETAIL_ARRAY_UTILS_H_
#ifndef TOPI_DETAIL_ARRAY_UTILS_H_
#define TOPI_DETAIL_ARRAY_UTILS_H_
#define TOPI_DETAIL_ARRAY_UTILS_H_
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
namespace
topi
{
namespace
topi
{
namespace
detail
{
namespace
detail
{
...
...
topi/include/topi/detail/broadcast.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -30,7 +30,8 @@
...
@@ -30,7 +30,8 @@
#include <string>
#include <string>
#include "tvm/ir_pass.h"
#include "tvm/ir_pass.h"
#include "tvm/tvm.h"
#include "tvm/operation.h"
#include "tvm/expr_operator.h"
#include "topi/detail/constant_utils.h"
#include "topi/detail/constant_utils.h"
namespace
topi
{
namespace
topi
{
...
...
topi/include/topi/detail/constant_utils.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -28,7 +28,7 @@
...
@@ -28,7 +28,7 @@
#include <string>
#include <string>
#include <vector>
#include <vector>
#include "tvm/
tvm
.h"
#include "tvm/
expr
.h"
#include "tvm/ir_pass.h"
#include "tvm/ir_pass.h"
namespace
topi
{
namespace
topi
{
...
...
topi/include/topi/detail/extern.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,14 +18,13 @@
...
@@ -18,14 +18,13 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file detail/extern.h
* \file detail/extern.h
* \brief Helpers for using external functions
* \brief Helpers for using external functions
*/
*/
#ifndef TOPI_DETAIL_EXTERN_H_
#ifndef TOPI_DETAIL_EXTERN_H_
#define TOPI_DETAIL_EXTERN_H_
#define TOPI_DETAIL_EXTERN_H_
#include <tvm/
tvm
.h>
#include <tvm/
operation
.h>
#include <vector>
#include <vector>
#include <string>
#include <string>
...
...
topi/include/topi/detail/fuse.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,14 +18,13 @@
...
@@ -18,14 +18,13 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file fuse.h
* \file fuse.h
* \brief Fuse operation
* \brief Fuse operation
*/
*/
#ifndef TOPI_DETAIL_FUSE_H_
#ifndef TOPI_DETAIL_FUSE_H_
#define TOPI_DETAIL_FUSE_H_
#define TOPI_DETAIL_FUSE_H_
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
namespace
topi
{
namespace
topi
{
namespace
detail
{
namespace
detail
{
...
...
topi/include/topi/detail/pad_utils.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -27,7 +27,8 @@
...
@@ -27,7 +27,8 @@
#include <vector>
#include <vector>
#include "tvm/tvm.h"
#include "tvm/expr.h"
#include "tvm/expr_operator.h"
namespace
topi
{
namespace
topi
{
namespace
detail
{
namespace
detail
{
...
...
topi/include/topi/detail/ravel_unravel.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -27,7 +27,8 @@
...
@@ -27,7 +27,8 @@
#include <vector>
#include <vector>
#include "tvm/tvm.h"
#include "tvm/operation.h"
#include "tvm/expr_operator.h"
namespace
topi
{
namespace
topi
{
namespace
detail
{
namespace
detail
{
...
...
topi/include/topi/elemwise.h
View file @
025a6c80
...
@@ -28,7 +28,6 @@
...
@@ -28,7 +28,6 @@
#include <string>
#include <string>
#include "topi/tags.h"
#include "topi/tags.h"
#include "tvm/tvm.h"
#include "tvm/ir.h"
#include "tvm/ir.h"
#include "tvm/ir_pass.h"
#include "tvm/ir_pass.h"
#include "broadcast.h"
#include "broadcast.h"
...
...
topi/include/topi/generic/default.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,16 +18,15 @@
...
@@ -18,16 +18,15 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file generic/default.h
* \file generic/default.h
* \brief Generic default schedule
* \brief Generic default schedule
*/
*/
#ifndef TOPI_GENERIC_DEFAULT_H_
#ifndef TOPI_GENERIC_DEFAULT_H_
#define TOPI_GENERIC_DEFAULT_H_
#define TOPI_GENERIC_DEFAULT_H_
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/fuse.h"
#include "topi/detail/fuse.h"
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/build_module.h"
#include "tvm/build_module.h"
namespace
topi
{
namespace
topi
{
...
...
topi/include/topi/generic/extern.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,16 +18,15 @@
...
@@ -18,16 +18,15 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file generic/extern.h
* \file generic/extern.h
* \brief Schedule for extern followed by injective ops
* \brief Schedule for extern followed by injective ops
*/
*/
#ifndef TOPI_GENERIC_EXTERN_H_
#ifndef TOPI_GENERIC_EXTERN_H_
#define TOPI_GENERIC_EXTERN_H_
#define TOPI_GENERIC_EXTERN_H_
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/fuse.h"
#include "topi/detail/fuse.h"
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/build_module.h"
#include "tvm/build_module.h"
namespace
topi
{
namespace
topi
{
...
...
topi/include/topi/generic/injective.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,16 +18,15 @@
...
@@ -18,16 +18,15 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file generic/injective.h
* \file generic/injective.h
* \brief Generic schedule for injective operations
* \brief Generic schedule for injective operations
*/
*/
#ifndef TOPI_GENERIC_INJECTIVE_H_
#ifndef TOPI_GENERIC_INJECTIVE_H_
#define TOPI_GENERIC_INJECTIVE_H_
#define TOPI_GENERIC_INJECTIVE_H_
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/fuse.h"
#include "topi/detail/fuse.h"
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/build_module.h"
#include "tvm/build_module.h"
namespace
topi
{
namespace
topi
{
...
...
topi/include/topi/image/resize.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -34,7 +34,8 @@
...
@@ -34,7 +34,8 @@
#include "topi/elemwise.h"
#include "topi/elemwise.h"
#include "topi/detail/ravel_unravel.h"
#include "topi/detail/ravel_unravel.h"
#include "topi/detail/constant_utils.h"
#include "topi/detail/constant_utils.h"
#include "tvm/tvm.h"
#include "tvm/operation.h"
#include "tvm/expr_operator.h"
namespace
topi
{
namespace
topi
{
namespace
image
{
namespace
image
{
...
...
topi/include/topi/nn.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -32,7 +32,8 @@
...
@@ -32,7 +32,8 @@
#include "topi/detail/constant_utils.h"
#include "topi/detail/constant_utils.h"
#include "tvm/ir.h"
#include "tvm/ir.h"
#include "tvm/ir_pass.h"
#include "tvm/ir_pass.h"
#include "tvm/tvm.h"
#include "tvm/operation.h"
#include "tvm/expr_operator.h"
namespace
topi
{
namespace
topi
{
using
namespace
tvm
;
using
namespace
tvm
;
...
...
topi/include/topi/nn/batch_matmul.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -28,7 +28,7 @@
...
@@ -28,7 +28,7 @@
#include <string>
#include <string>
#include "topi/tags.h"
#include "topi/tags.h"
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
namespace
topi
{
namespace
topi
{
namespace
nn
{
namespace
nn
{
...
...
topi/include/topi/nn/bias_add.h
View file @
025a6c80
...
@@ -30,7 +30,8 @@
...
@@ -30,7 +30,8 @@
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/broadcast.h"
#include "topi/broadcast.h"
#include "topi/transform.h"
#include "topi/transform.h"
#include "tvm/tvm.h"
#include "tvm/operation.h"
#include "tvm/expr_operator.h"
namespace
topi
{
namespace
topi
{
namespace
nn
{
namespace
nn
{
...
...
topi/include/topi/nn/bnn.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,7 +18,6 @@
...
@@ -18,7 +18,6 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \brief Binary op constructions
* \brief Binary op constructions
* \file nn/bnn.h
* \file nn/bnn.h
*/
*/
...
@@ -27,7 +26,7 @@
...
@@ -27,7 +26,7 @@
#include <string>
#include <string>
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/ir_pass.h"
#include "tvm/ir_pass.h"
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/constant_utils.h"
#include "topi/detail/constant_utils.h"
...
...
topi/include/topi/nn/dense.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -28,7 +28,7 @@
...
@@ -28,7 +28,7 @@
#include <string>
#include <string>
#include "topi/tags.h"
#include "topi/tags.h"
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
namespace
topi
{
namespace
topi
{
namespace
nn
{
namespace
nn
{
...
...
topi/include/topi/nn/dilate.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -27,7 +27,7 @@
...
@@ -27,7 +27,7 @@
#include <string>
#include <string>
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/ir_pass.h"
#include "tvm/ir_pass.h"
#include "topi/tags.h"
#include "topi/tags.h"
...
...
topi/include/topi/nn/flatten.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -30,7 +30,9 @@
...
@@ -30,7 +30,9 @@
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/constant_utils.h"
#include "topi/detail/constant_utils.h"
#include "tvm/tvm.h"
#include "tvm/operation.h"
#include "tvm/expr_operator.h"
namespace
topi
{
namespace
topi
{
namespace
nn
{
namespace
nn
{
...
...
topi/include/topi/nn/l2_normalize.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,7 +18,6 @@
...
@@ -18,7 +18,6 @@
*/
*/
/*!
/*!
* Copyright (c) 2018 by Contributors
* \brief l2 normalization op constructions
* \brief l2 normalization op constructions
* \file nn/l2_normalize.h
* \file nn/l2_normalize.h
*/
*/
...
@@ -28,7 +27,7 @@
...
@@ -28,7 +27,7 @@
#include <string>
#include <string>
#include <algorithm>
#include <algorithm>
#include "topi/tags.h"
#include "topi/tags.h"
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
namespace
topi
{
namespace
topi
{
namespace
nn
{
namespace
nn
{
using
namespace
tvm
;
using
namespace
tvm
;
...
...
topi/include/topi/nn/local_response_norm.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,7 +18,6 @@
...
@@ -18,7 +18,6 @@
*/
*/
/*!
/*!
* Copyright (c) 2018 by Contributors
* \brief local response normalization op constructions
* \brief local response normalization op constructions
* \file nn/local_response_norm.h
* \file nn/local_response_norm.h
*/
*/
...
@@ -28,7 +27,7 @@
...
@@ -28,7 +27,7 @@
#include <string>
#include <string>
#include "topi/tags.h"
#include "topi/tags.h"
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
namespace
topi
{
namespace
topi
{
namespace
nn
{
namespace
nn
{
...
...
topi/include/topi/nn/mapping.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,7 +18,6 @@
...
@@ -18,7 +18,6 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \brief Mapping op constructions
* \brief Mapping op constructions
* \file nn/mapping.h
* \file nn/mapping.h
*/
*/
...
@@ -28,7 +27,7 @@
...
@@ -28,7 +27,7 @@
#include <string>
#include <string>
#include "topi/tags.h"
#include "topi/tags.h"
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
namespace
topi
{
namespace
topi
{
namespace
nn
{
namespace
nn
{
...
...
topi/include/topi/nn/pooling.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -28,7 +28,6 @@
...
@@ -28,7 +28,6 @@
#include <string>
#include <string>
#include <vector>
#include <vector>
#include "tvm/tvm.h"
#include "tvm/ir_pass.h"
#include "tvm/ir_pass.h"
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/pad_utils.h"
#include "topi/detail/pad_utils.h"
...
...
topi/include/topi/nn/softmax.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -30,7 +30,8 @@
...
@@ -30,7 +30,8 @@
#include "topi/reduction.h"
#include "topi/reduction.h"
#include "topi/tags.h"
#include "topi/tags.h"
#include "tvm/tvm.h"
#include "tvm/operation.h"
#include "tvm/expr_operator.h"
namespace
topi
{
namespace
topi
{
namespace
nn
{
namespace
nn
{
...
...
topi/include/topi/reduction.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -36,7 +36,8 @@
...
@@ -36,7 +36,8 @@
#include "topi/transform.h"
#include "topi/transform.h"
#include "topi/detail/ravel_unravel.h"
#include "topi/detail/ravel_unravel.h"
#include "topi/detail/constant_utils.h"
#include "topi/detail/constant_utils.h"
#include "tvm/tvm.h"
#include "tvm/operation.h"
#include "tvm/expr_operator.h"
namespace
topi
{
namespace
topi
{
...
...
topi/include/topi/rocm/dense.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,14 +18,13 @@
...
@@ -18,14 +18,13 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file rocm/dense.h
* \file rocm/dense.h
* \brief rocm schedule for dense operation
* \brief rocm schedule for dense operation
*/
*/
#ifndef TOPI_ROCM_DENSE_H_
#ifndef TOPI_ROCM_DENSE_H_
#define TOPI_ROCM_DENSE_H_
#define TOPI_ROCM_DENSE_H_
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/build_module.h"
#include "tvm/build_module.h"
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/array_utils.h"
#include "topi/detail/array_utils.h"
...
@@ -100,4 +99,3 @@ inline Schedule schedule_dense(const Target &target, const Array<Tensor>& outs)
...
@@ -100,4 +99,3 @@ inline Schedule schedule_dense(const Target &target, const Array<Tensor>& outs)
}
// namespace rocm
}
// namespace rocm
}
// namespace topi
}
// namespace topi
#endif // TOPI_ROCM_DENSE_H_
#endif // TOPI_ROCM_DENSE_H_
topi/include/topi/rocm/normalization.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,14 +18,13 @@
...
@@ -18,14 +18,13 @@
*/
*/
/*!
/*!
* Copyright (c) 2018 by Contributors
* \file rocm/normalization.h
* \file rocm/normalization.h
* \brief rocm schedule for LRN and l2 normalization operations
* \brief rocm schedule for LRN and l2 normalization operations
*/
*/
#ifndef TOPI_ROCM_NORMALIZATION_H_
#ifndef TOPI_ROCM_NORMALIZATION_H_
#define TOPI_ROCM_NORMALIZATION_H_
#define TOPI_ROCM_NORMALIZATION_H_
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/build_module.h"
#include "tvm/build_module.h"
#include "topi/tags.h"
#include "topi/tags.h"
...
...
topi/include/topi/transform.h
View file @
025a6c80
...
@@ -35,7 +35,8 @@
...
@@ -35,7 +35,8 @@
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/ravel_unravel.h"
#include "topi/detail/ravel_unravel.h"
#include "topi/detail/constant_utils.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"
#include "tvm/data_layout.h"
namespace
topi
{
namespace
topi
{
...
...
topi/include/topi/vision/reorg.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -32,7 +32,8 @@
...
@@ -32,7 +32,8 @@
#include "topi/reduction.h"
#include "topi/reduction.h"
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/transform.h"
#include "topi/transform.h"
#include "tvm/tvm.h"
#include "tvm/operation.h"
#include "tvm/expr_operator.h"
namespace
topi
{
namespace
topi
{
namespace
vision
{
namespace
vision
{
...
...
topi/include/topi/x86/bnn.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,16 +18,15 @@
...
@@ -18,16 +18,15 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file x86/bnn.h
* \file x86/bnn.h
* \brief x86 schedule for binary operations
* \brief x86 schedule for binary operations
*/
*/
#ifndef TOPI_X86_BNN_H_
#ifndef TOPI_X86_BNN_H_
#define TOPI_X86_BNN_H_
#define TOPI_X86_BNN_H_
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/fuse.h"
#include "topi/detail/fuse.h"
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/build_module.h"
#include "tvm/build_module.h"
namespace
topi
{
namespace
topi
{
...
...
topi/include/topi/x86/default.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,16 +18,15 @@
...
@@ -18,16 +18,15 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file x86/default.h
* \file x86/default.h
* \brief default x86 schedule
* \brief default x86 schedule
*/
*/
#ifndef TOPI_X86_DEFAULT_H_
#ifndef TOPI_X86_DEFAULT_H_
#define TOPI_X86_DEFAULT_H_
#define TOPI_X86_DEFAULT_H_
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/fuse.h"
#include "topi/detail/fuse.h"
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/build_module.h"
#include "tvm/build_module.h"
namespace
topi
{
namespace
topi
{
...
...
topi/include/topi/x86/injective.h
View file @
025a6c80
...
@@ -6,9 +6,9 @@
...
@@ -6,9 +6,9 @@
* to you under the Apache License, Version 2.0 (the
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
* with the License. You may obtain a copy of the License at
*
*
* http://www.apache.org/licenses/LICENSE-2.0
* http://www.apache.org/licenses/LICENSE-2.0
*
*
* Unless required by applicable law or agreed to in writing,
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
...
@@ -18,16 +18,15 @@
...
@@ -18,16 +18,15 @@
*/
*/
/*!
/*!
* Copyright (c) 2017 by Contributors
* \file x86/injective.h
* \file x86/injective.h
* \brief x86 schedule for injective ops
* \brief x86 schedule for injective ops
*/
*/
#ifndef TOPI_X86_INJECTIVE_H_
#ifndef TOPI_X86_INJECTIVE_H_
#define TOPI_X86_INJECTIVE_H_
#define TOPI_X86_INJECTIVE_H_
#include "topi/tags.h"
#include "topi/tags.h"
#include "topi/detail/fuse.h"
#include "topi/detail/fuse.h"
#include "tvm/
tvm
.h"
#include "tvm/
operation
.h"
#include "tvm/build_module.h"
#include "tvm/build_module.h"
namespace
topi
{
namespace
topi
{
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment