struct_info.h File Reference

#include <tvm/ir/env_func.h>
#include <tvm/ir/source_map.h>
#include <tvm/node/node.h>
#include <tvm/relax/block_builder.h>
#include <tvm/relax/expr.h>
#include <tvm/relax/type.h>

Include dependency graph for struct_info.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	tvm::relax::ObjectStructInfoNode
	Opaque object. More...
class	tvm::relax::ObjectStructInfo
	Managed reference to ObjectStructInfoNode. More...
class	tvm::relax::PrimStructInfoNode
	Primitive value. More...
class	tvm::relax::PrimStructInfo
	Managed reference to PrimStructInfoNode. More...
class	tvm::relax::ShapeStructInfoNode
	StructInfo of shape value. More...
class	tvm::relax::ShapeStructInfo
	Managed reference to ShapeStructInfoNode. More...
class	tvm::relax::TensorStructInfoNode
	StructInfo of Tensor. More...
class	tvm::relax::TensorStructInfo
	Managed reference to TensorStructInfoNode. More...
class	tvm::relax::TupleStructInfoNode
	StructInfo of Tuple. More...
class	tvm::relax::TupleStructInfo
	Managed reference to TupleStructInfoNode. More...
class	tvm::relax::FuncStructInfoNode
	Structure information about function. More...
class	tvm::relax::FuncStructInfo
	Managed reference to FuncStructInfoNode. More...

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::relax

Typedefs
using	tvm::relax::StructInfoDeriveFunc = TypedEnvFunc< StructInfo(const Call &call, const BlockBuilder &ctx)>
	custom-defined StructInfo derivation function. More...

Functions
template<typename T >
Optional< T >	tvm::relax::MatchStructInfo (const Expr &expr)
	Match and check if expr have StructInfo T and return it. More...
template<typename T >
const T *	tvm::relax::GetStructInfoAs (const Expr &expr)
	Get the structure info of a given expr and try to cast it as const T*. More...
StructInfo	tvm::relax::GetStructInfo (const Expr &expr)
	Get the underlying structure info of expr. More...
bool	tvm::relax::HasVoidStructInfo (const Expr &expr)
	Whether the expr has void struct info. More...
void	tvm::relax::UpdateStructInfo (Expr expr, StructInfo struct_info)
	Update the struct info of an Expr. More...