tvm
Loading...
Searching...
No Matches
Classes | Namespaces | Typedefs | Functions
type.h File Reference

Relax types, including the richer dependent Relax type nodes. More...

#include <tvm/ffi/function.h>
#include <tvm/ffi/reflection/registry.h>
#include <tvm/ir/attrs.h>
#include <tvm/ir/env_func.h>
#include <tvm/ir/global_info.h>
#include <tvm/ir/prim/expr.h>
#include <tvm/ir/type.h>
#include <string>
#include <utility>
Include dependency graph for type.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  tvm::relax::PackedFuncTypeNode
 
class  tvm::relax::PackedFuncType
 
class  tvm::relax::AnyTypeNode
 Base type of all Relax type information. More...
 
class  tvm::relax::AnyType
 Managed reference to AnyTypeNode. More...
 
class  tvm::relax::ShapeTypeNode
 Type of shape value. More...
 
class  tvm::relax::ShapeType
 Managed reference to ShapeTypeNode. More...
 
class  tvm::relax::TensorTypeNode
 Type of Tensor. More...
 
class  tvm::relax::TensorType
 Managed reference to TensorTypeNode. More...
 
class  tvm::relax::FuncTypeNode
 Function type information. More...
 
class  tvm::relax::FuncType
 Managed reference to FuncTypeNode. More...
 

Namespaces

namespace  tvm
 An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
 
namespace  tvm::relax
 

Typedefs

using tvm::relax::Expr = tvm::Expr
 
using tvm::relax::ExprNode = tvm::ExprNode
 
using tvm::relax::ObjectTypeNode = AnyTypeNode
 
using tvm::relax::ObjectType = AnyType
 
using tvm::relax::TypeDeriveFunc = TypedEnvFunc< Type(const Call &call, const BlockBuilder &ctx)>
 custom-defined Type derivation function.
 

Functions

template<typename T >
ffi::Optional< Ttvm::relax::MatchType (const Expr &expr)
 Match and check if expr has Relax type T and return it.
 
template<typename T >
const Ttvm::relax::GetTypeAs (const Expr &expr)
 Get the type of a given expr and try to cast it as const T*.
 
Type tvm::relax::GetType (const Expr &expr)
 Get the underlying Relax type of expr.
 
bool tvm::relax::HasVoidType (const Expr &expr)
 Whether the expr has void type.
 
void tvm::relax::UpdateType (Expr expr, Type ty)
 Update the type of an Expr.
 

Detailed Description

Relax types, including the richer dependent Relax type nodes.