tvm::relax::attr Namespace Reference

Variables
constexpr const char *	kPrimitive = "Primitive"
	Mark the function as a primitive function.
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.
constexpr const char *	kComposite = "Composite"
	Treat the function as a composite operator.
constexpr const char *	kPartitionedFromPattern = "PartitionedFromPattern"
	Indicate the function was created by the Pattern Partitioning Pass.
constexpr const char *	kWorkspaceSize = "WorkspaceSize"
	The required workspace for an external function.
constexpr const char *	kForcePure = "relax.force_pure"
	Override checking purity for this function and treat as pure (is_pure must be set to true)
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.
constexpr const char *	kOpPattern = "op_pattern"
constexpr const char *	kOpAttrs = "op_attrs"
constexpr const char *	kOperatorName = "operator_name"
constexpr const char *	kLibraryKernel = "library_kernel"
constexpr const char *	kCSource = "c_source"
constexpr const char *	kCSourceFmt = "c_source_fmt"
constexpr const char *	kTirVarUpperBound = "tir_var_upper_bound"
constexpr const char *	kTirVarLowerBound = "tir_var_lower_bound"
constexpr const char *	kTirNonNegativeVar = "tir_non_negative_var"
constexpr const char *	kMemoryPlanDynamicFuncOutput = "relax.memory_plan_dynamic_func_output"
constexpr const char *	kCaptureSymbolicVars = "relax.rewrite_cuda_graph.capture_symbolic_vars"
constexpr const char *	kScoped = "scoped"

Variable Documentation

kCaptureSymbolicVars

constexpr const char* tvm::relax::attr::kCaptureSymbolicVars = "relax.rewrite_cuda_graph.capture_symbolic_vars"

constexpr

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.

kCSource

constexpr const char* tvm::relax::attr::kCSource = "c_source"

constexpr

kCSourceFmt

constexpr const char* tvm::relax::attr::kCSourceFmt = "c_source_fmt"

constexpr

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)

kLibraryKernel

constexpr const char* tvm::relax::attr::kLibraryKernel = "library_kernel"

constexpr

kMemoryPlanDynamicFuncOutput

constexpr const char* tvm::relax::attr::kMemoryPlanDynamicFuncOutput = "relax.memory_plan_dynamic_func_output"

constexpr

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.

kOpAttrs

constexpr const char* tvm::relax::attr::kOpAttrs = "op_attrs"

constexpr

kOperatorName

constexpr const char* tvm::relax::attr::kOperatorName = "operator_name"

constexpr

kOpPattern

constexpr const char* tvm::relax::attr::kOpPattern = "op_pattern"

constexpr

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.

kScoped

constexpr const char* tvm::relax::attr::kScoped = "scoped"

constexpr

kTirNonNegativeVar

constexpr const char* tvm::relax::attr::kTirNonNegativeVar = "tir_non_negative_var"

constexpr

kTirVarLowerBound

constexpr const char* tvm::relax::attr::kTirVarLowerBound = "tir_var_lower_bound"

constexpr

kTirVarUpperBound

constexpr const char* tvm::relax::attr::kTirVarUpperBound = "tir_var_upper_bound"

constexpr

kWorkspaceSize

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

constexpr

The required workspace for an external function.