tvm
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/type.h>
#include <tvm/tirx/expr.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

 tvm
 An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
 
 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. More...
 

Functions

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

Detailed Description

Relax types, including the richer dependent Relax type nodes.