|
tvm
|
Managed reference to ScheduleContextNode. More...
#include <tirx_op.h>
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) | |
Managed reference to ScheduleContextNode.
| 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.
| target | The target of the kernel. |
| exec_scope | The exec scope of the operator. |
| launch_params | The kernel launch parameters. |
| var_range_map | A map from loop variables to their ranges. |
| alloc_only | Whether the schedule context is only used for buffer allocation. |
| callbacks | The callbacks to be handled when the operator is scheduled. |
| tvm::tirx::ScheduleContext::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE | ( | ScheduleContext | , |
| ffi::ObjectRef | , | ||
| ScheduleContextNode | |||
| ) |