tvm
Public Member Functions | List of all members
tvm::tirx::ScheduleContext Class Reference

Managed reference to ScheduleContextNode. More...

#include <tirx_op.h>

Inheritance diagram for tvm::tirx::ScheduleContext:
Collaboration diagram for tvm::tirx::ScheduleContext:

Public Member Functions

 ScheduleContext (Target target, ExecScope exec_scope, ffi::Map< ffi::String, IterVar > launch_params={}, ffi::Map< Var, Range > var_range_map={}, bool alloc_only=false, ffi::Map< ffi::String, ffi::ObjectRef > callbacks={})
 Constructor. More...
 
 TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (ScheduleContext, ffi::ObjectRef, ScheduleContextNode)
 

Detailed Description

Managed reference to ScheduleContextNode.

Constructor & Destructor Documentation

◆ ScheduleContext()

tvm::tirx::ScheduleContext::ScheduleContext ( Target  target,
ExecScope  exec_scope,
ffi::Map< ffi::String, IterVar launch_params = {},
ffi::Map< Var, Range var_range_map = {},
bool  alloc_only = false,
ffi::Map< ffi::String, ffi::ObjectRef >  callbacks = {} 
)

Constructor.

Parameters
targetThe target of the kernel.
exec_scopeThe exec scope of the operator.
launch_paramsThe kernel launch parameters.
var_range_mapA map from loop variables to their ranges.
alloc_onlyWhether the schedule context is only used for buffer allocation.
callbacksThe callbacks to be handled when the operator is scheduled.

Member Function Documentation

◆ TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE()

tvm::tirx::ScheduleContext::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE ( ScheduleContext  ,
ffi::ObjectRef  ,
ScheduleContextNode   
)

The documentation for this class was generated from the following file: