tvm::relay::attr Namespace Reference

namespace of the attributes that can be attached to a relay::Function. More...

Variables
constexpr const char *	kPrimitive = "Primitive"
	Mark the function as representing a sub-graph which is to be lowered or compiled as a unit. For example, the function may represent a kernel which TVM will lower to a PrimFunc. If present should be bound to Integer(1). May be accompanied by "Compiler", see below. The function body should be considered opaque by Relay, and many passes simply ignore these functions. More...
constexpr const char *	kExtern = "Extern"
	Mark the function as externally implemented, ie bound in a runtime::Module within the IRModule's "external_mods" attribute. If present should be bound to Integer(1). Generally the only attribute when present. More...
constexpr const char *	kCompiler = "Compiler"
	Indicates the name of the external codegen 'compiler' that should be used to lower or compile the function other than TVM's default lowering pipeline. The name may correspond to a TargetKind name. There may be a global function registered under 'relay.ext.{name}'. More...
constexpr const char *	kClosure = "Closure"
	Indicate if the function is a closure. More...
constexpr const char *	kParams = "__params__"
	Store a Var to parameter/Constant mapping on a Function. More...
constexpr const char *	kSkipOptimization = "SkipOptimization"
	Mark if the function should be avoided being optimized. More...
constexpr const char *	kComposite = "Composite"
	Treat the function as a composite operator. More...
constexpr const char *	kInline = "Inline"
	Mark the function to be inlined. More...
constexpr const char *	kPartitionedFromPattern = "PartitionedFromPattern"
	Indicate the function was created by the Pattern Partitioning Pass. More...
constexpr const char *	kReshapeOnly = "relay.reshape_only"
	Mark the function as only composed of reshape operations. More...

Detailed Description

namespace of the attributes that can be attached to a relay::Function.

Variable Documentation

kClosure

constexpr const char* tvm::relay::attr::kClosure = "Closure"

constexpr

Indicate if the function is a closure.

kCompiler

constexpr const char* tvm::relay::attr::kCompiler = "Compiler"

constexpr

Indicates the name of the external codegen 'compiler' that should be used to lower or compile the function other than TVM's default lowering pipeline. The name may correspond to a TargetKind name. There may be a global function registered under 'relay.ext.{name}'.

Type: String

kComposite

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

constexpr

Treat the function as a composite operator.

kExtern

constexpr const char* tvm::relay::attr::kExtern = "Extern"

constexpr

Mark the function as externally implemented, ie bound in a runtime::Module within the IRModule's "external_mods" attribute. If present should be bound to Integer(1). Generally the only attribute when present.

Type: Integer

kInline

constexpr const char* tvm::relay::attr::kInline = "Inline"

constexpr

Mark the function to be inlined.

kParams

constexpr const char* tvm::relay::attr::kParams = "__params__"

constexpr

Store a Var to parameter/Constant mapping on a Function.

kPartitionedFromPattern

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

constexpr

Indicate the function was created by the Pattern Partitioning Pass.

kPrimitive

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

constexpr

Mark the function as representing a sub-graph which is to be lowered or compiled as a unit. For example, the function may represent a kernel which TVM will lower to a PrimFunc. If present should be bound to Integer(1). May be accompanied by "Compiler", see below. The function body should be considered opaque by Relay, and many passes simply ignore these functions.

Type: Integer

kReshapeOnly

constexpr const char* tvm::relay::attr::kReshapeOnly = "relay.reshape_only"

constexpr

Mark the function as only composed of reshape operations.

kSkipOptimization

constexpr const char* tvm::relay::attr::kSkipOptimization = "SkipOptimization"

constexpr

Mark if the function should be avoided being optimized.