feature.h File Reference

Detect features used in Expr/Module. More...

#include <tvm/ir/module.h>
#include <tvm/relay/expr.h>
#include <bitset>
#include <string>

Include dependency graph for feature.h:

Go to the source code of this file.

Classes
class	tvm::relay::FeatureSet
	A finite set of Feature. More...

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::relay
	Relay: a high level functional IR for TVM.

Enumerations
enum	tvm::relay::Feature : int {   tvm::relay::fVar = 0, tvm::relay::fGlobalVar = 1, tvm::relay::fConstant = 2, tvm::relay::fTuple = 3,   tvm::relay::fTupleGetItem = 4, tvm::relay::fFunction = 5, tvm::relay::fOp = 6, tvm::relay::fCall = 7,   tvm::relay::fLet = 8, tvm::relay::fIf = 9, tvm::relay::fRefCreate = 10, tvm::relay::fRefRead = 11,   tvm::relay::fRefWrite = 12, tvm::relay::fConstructor = 13, tvm::relay::fMatch = 14, tvm::relay::fGraph = 15,   tvm::relay::fLetRec = 16 }
	Different kinds of relay feature a program might use. More...

Functions
FeatureSet	tvm::relay::DetectFeature (const RelayExpr &expr)
	Calculate the feature of the program. More...
FeatureSet	tvm::relay::DetectFeature (const IRModule &mod)
	Calculate the feature of the program. More...
FeatureSet	tvm::relay::DetectFeature (const Expr &expr, const IRModule &mod)
	Calculate the feature of the program. More...
void	tvm::relay::CheckFeature (const RelayExpr &expr, const FeatureSet &fs)
	Check the feature of the program. More...
void	tvm::relay::CheckFeature (const IRModule &mod, const FeatureSet &fs)
	Check the feature of the program. More...
void	tvm::relay::CheckFeature (const RelayExpr &expr, const IRModule &mod, const FeatureSet &fs)
	Check the feature of the program. More...

Variables
constexpr size_t	tvm::relay::feature_count = 17

Detailed Description

Detect features used in Expr/Module.