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 20 /*! 21 * \file tvm/relay/type.h 22 * \brief Relay typed AST nodes. 23 */ 24 #ifndef TVM_RELAY_TYPE_H_ 25 #define TVM_RELAY_TYPE_H_ 26 27 #include <tvm/ir/attrs.h> 28 #include <tvm/ir/env_func.h> 29 #include <tvm/ir/tensor_type.h> 30 #include <tvm/ir/type.h> 31 #include <tvm/ir/type_relation.h> 32 #include <tvm/runtime/registry.h> 33 #include <tvm/tir/expr.h> 34 35 #include <string> 36 37 #include "base.h" 38 39 namespace tvm { 40 namespace relay { 41 42 // namespace update for backward compact 43 // will be removed later. 44 using AnyNode = tvm::tir::AnyNode; 45 using Any = tvm::tir::Any; 46 using Kind = TypeKind; 47 using Type = tvm::Type; 48 using TypeNode = tvm::TypeNode; 49 using TypeVar = tvm::TypeVar; 50 using TypeVarNode = tvm::TypeVarNode; 51 using GlobalTypeVar = tvm::GlobalTypeVar; 52 using GlobalTypeVarNode = tvm::GlobalTypeVarNode; 53 using TupleType = tvm::TupleType; 54 using TupleTypeNode = tvm::TupleTypeNode; 55 using TypeConstraint = tvm::TypeConstraint; 56 using TypeConstraintNode = tvm::TypeConstraintNode; 57 using FuncType = tvm::FuncType; 58 using FuncTypeNode = tvm::FuncTypeNode; 59 using IncompleteType = tvm::IncompleteType; 60 using IncompleteTypeNode = tvm::IncompleteTypeNode; 61 using RelayRefType = tvm::RelayRefType; 62 using RelayRefTypeNode = tvm::RelayRefTypeNode; 63 using TensorType = tvm::TensorType; 64 using TensorTypeNode = tvm::TensorTypeNode; 65 using TypeCall = tvm::TypeCall; 66 using TypeCallNode = tvm::TypeCallNode; 67 using TypeRelation = tvm::TypeRelation; 68 using TypeRelationNode = tvm::TypeRelationNode; 69 using TypeRelationFn = tvm::TypeRelationFn; 70 using TypeReporter = tvm::TypeReporter; 71 using TypeReporterNode = tvm::TypeReporterNode; 72 73 } // namespace relay 74 } // namespace tvm 75 #endif // TVM_RELAY_TYPE_H_ 76