Class FunctionFrameObj

Class FunctionFrameObj#

Defined in File frame.h

Inheritance Relationships#

Base Type#

public tvm::script::ir_builder::ir::StmtFrameObj (Class StmtFrameObj)

Class Documentation#

class FunctionFrameObj : public tvm::script::ir_builder::ir::StmtFrameObj #

A frame that represents the Function containing TIR statements.

See also

Public Types

using AttrValidator = ffi::TypedFunction<void(const FunctionFrameObj*, const ffi::Any&)>#: Register validation for an extension-owned function attribute.

Public Functions

virtual void ExitWithScope() final#

The method called when exiting RAII scope.

See also

tvm::support::With

void ValidateAttrs() const#

virtual tvm::tirx::Function FinalizeFunction(tvm::tirx::Function func)#: Complete dialect-specific function construction before publication.

Public Members

ffi::Optional<ffi::String> name#: The name of the block.

ffi::Array<tvm::Var> args#: Function parameters.

bool is_private#: Whether the Function is annotated as private.

ffi::Optional<Type> ret_type#: The return type of the function.

ffi::Map<ffi::String, Any> attrs#: Additional attributes storing the meta-data.

bool persistent#: Whether it is a persistent kernel.

bool is_declaration = {false}#: Whether this frame collects a signature for later definition.

ffi::Optional<tvm::tirx::Function> function#: Finalized function and its module identity.

ffi::Optional<GlobalVar> global_var#

Public Static Functions

static inline void RegisterReflection()#

static inline int32_t _GetOrAllocRuntimeTypeIndex()#

static inline int32_t RuntimeTypeIndex() noexcept#

static void RegisterAttrValidator(ffi::String key, AttrValidator validator)#