tvm
backend.h
Go to the documentation of this file.
1 /*
2  * Licensed to the Apache Software Foundation (ASF) under one
3  * or more contributor license agreements. See the NOTICE file
4  * distributed with this work for additional information
5  * regarding copyright ownership. The ASF licenses this file
6  * to you under the Apache License, Version 2.0 (the
7  * "License"); you may not use this file except in compliance
8  * with the License. You may obtain a copy of the License at
9  *
10  * http://www.apache.org/licenses/LICENSE-2.0
11  *
12  * Unless required by applicable law or agreed to in writing,
13  * software distributed under the License is distributed on an
14  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15  * KIND, either express or implied. See the License for the
16  * specific language governing permissions and limitations
17  * under the License.
18  */
19 
24 #ifndef TVM_RELAX_BACKEND_H_
25 #define TVM_RELAX_BACKEND_H_
26 
27 #include <tvm/relax/transform.h>
28 
29 namespace tvm {
30 namespace relax {
31 namespace transform {
32 
39 
45 TVM_DLL Pass VMShapeLower();
46 
47 } // namespace transform
48 } // namespace relax
49 } // namespace tvm
50 
51 #endif // TVM_RELAX_BACKEND_H_
Definition: transform.h:426
Pass VMShapeLower()
Lower the shape expression in relax to VM shape heap and TIR functions.
Pass LowerRuntimeBuiltin()
Perform builtin lowering to map most of the op to VM builtin functions.
runtime implementation for LibTorch/TorchScript.
Definition: analyzer.h:36
Relax specific transformation passes.