tvm::relax::attr Namespace Reference

Variables
constexpr const char *	kPrimitive = "Primitive"
	Mark the function as a primitive function. More...
constexpr const char *	kCodegen = "Codegen"
	Indicate the codegen that should be used for building this function. When this is unset or set to "default", the default compilation pipeline will be used. More...
constexpr const char *	kComposite = "Composite"
	Treat the function as a composite operator. More...
constexpr const char *	kPartitionedFromPattern = "PartitionedFromPattern"
	Indicate the function was created by the Pattern Partitioning Pass. More...
constexpr const char *	kWorkspaceSize = "WorkspaceSize"
	The required workspace for an external function. More...
constexpr const char *	kForcePure = "relax.force_pure"
	Override checking purity for this function and treat as pure (is_pure must be set to true) More...
constexpr const char *	kNumInput = "num_input"
	The number of inputs of a function. If a function has the num_input attribute, the last func->params.size() - num_inputs arguments are assumed to be weights that are fixed across invocations. More...

Variable Documentation

kCodegen

constexpr const char* tvm::relax::attr::kCodegen = "Codegen"

constexpr

Indicate the codegen that should be used for building this function. When this is unset or set to "default", the default compilation pipeline will be used.

kComposite

constexpr const char* tvm::relax::attr::kComposite = "Composite"

constexpr

Treat the function as a composite operator.

kForcePure

constexpr const char* tvm::relax::attr::kForcePure = "relax.force_pure"

constexpr

Override checking purity for this function and treat as pure (is_pure must be set to true)

kNumInput

constexpr const char* tvm::relax::attr::kNumInput = "num_input"

constexpr

The number of inputs of a function. If a function has the num_input attribute, the last func->params.size() - num_inputs arguments are assumed to be weights that are fixed across invocations.

kPartitionedFromPattern

constexpr const char* tvm::relax::attr::kPartitionedFromPattern = "PartitionedFromPattern"

constexpr

Indicate the function was created by the Pattern Partitioning Pass.

kPrimitive

constexpr const char* tvm::relax::attr::kPrimitive = "Primitive"

constexpr

Mark the function as a primitive function.

kWorkspaceSize

constexpr const char* tvm::relax::attr::kWorkspaceSize = "WorkspaceSize"

constexpr

The required workspace for an external function.