tvm
Loading...
Searching...
No Matches
tvm::s_tir::ScheduleObj Member List

This is the complete list of members for tvm::s_tir::ScheduleObj, including all inherited members.

_type_mutabletvm::s_tir::ScheduleObjstatic
AddUnitLoop(const SBlockRV &block_rv)=0tvm::s_tir::ScheduleObjpure virtual
AddUnitLoop(const LoopRV &loop_rv)=0tvm::s_tir::ScheduleObjpure virtual
Annotate(const LoopRV &loop_rv, const ffi::String &ann_key, const Any &ann_val)=0tvm::s_tir::ScheduleObjpure virtual
Annotate(const SBlockRV &block_rv, const ffi::String &ann_key, const Any &ann_val)=0tvm::s_tir::ScheduleObjpure virtual
AnnotateTensorAccess(const SBlockRV &block_rv, int tensor_index, TensorIndexType tensor_index_type, const IndexMap &index_map)=0tvm::s_tir::ScheduleObjpure virtual
Bind(const LoopRV &loop_rv, const ffi::String &thread_axis)=0tvm::s_tir::ScheduleObjpure virtual
Blockize(const LoopRV &loop_rv, bool preserve_unit_iters=true)=0tvm::s_tir::ScheduleObjpure virtual
Blockize(const ffi::Array< SBlockRV > &blocks, bool preserve_unit_iters=true)=0tvm::s_tir::ScheduleObjpure virtual
CacheIndex(const SBlockRV &block_rv, const ffi::String &storage_scope, int cse_thresh)=0tvm::s_tir::ScheduleObjpure virtual
CacheInplace(const SBlockRV &block_rv, int read_tensor_index, const ffi::String &storage_scope)=0tvm::s_tir::ScheduleObjpure virtual
CacheRead(const SBlockRV &block_rv, int read_tensor_index, const ffi::String &storage_scope, const ffi::Array< SBlockRV > consumer_blocks={})=0tvm::s_tir::ScheduleObjpure virtual
CacheWrite(const SBlockRV &block_rv, int write_tensor_index, const ffi::String &storage_scope, const ffi::Array< SBlockRV > consumer_blocks={})=0tvm::s_tir::ScheduleObjpure virtual
ComputeAt(const SBlockRV &block_rv, const LoopRV &loop_rv, bool preserve_unit_loops, int index=-1)=0tvm::s_tir::ScheduleObjpure virtual
ComputeInline(const SBlockRV &block)=0tvm::s_tir::ScheduleObjpure virtual
Copy()=0tvm::s_tir::ScheduleObjpure virtual
DecomposePadding(const SBlockRV &block_rv, const LoopRV &loop_rv)=0tvm::s_tir::ScheduleObjpure virtual
DecomposeReduction(const SBlockRV &block_rv, const LoopRV &loop_rv)=0tvm::s_tir::ScheduleObjpure virtual
EnterPostproc()=0tvm::s_tir::ScheduleObjpure virtual
ForkSeed()=0tvm::s_tir::ScheduleObjpure virtual
func_working_on() const =0tvm::s_tir::ScheduleObjpure virtual
Fuse(const ffi::Array< LoopRV > &loop_rvs, bool preserve_unit_iters=true)=0tvm::s_tir::ScheduleObjpure virtual
FuseReductionEpilogue(const SBlockRV &reduction_block, const SBlockRV &epilogue_block)=0tvm::s_tir::ScheduleObjpure virtual
Get(const SBlockRV &block_rv) const =0tvm::s_tir::ScheduleObjpure virtual
Get(const LoopRV &loop_rv) const =0tvm::s_tir::ScheduleObjpure virtual
Get(const ExprRV &expr_rv) const =0tvm::s_tir::ScheduleObjpure virtual
GetChildBlocks(const SBlockRV &block_rv)=0tvm::s_tir::ScheduleObjpure virtual
GetChildBlocks(const LoopRV &loop_rv)=0tvm::s_tir::ScheduleObjpure virtual
GetConsumers(const SBlockRV &block_rv)=0tvm::s_tir::ScheduleObjpure virtual
GetLoops(const SBlockRV &block_rv)=0tvm::s_tir::ScheduleObjpure virtual
GetOutputBlocks(const SBlockRV &scope_block_rv)=0tvm::s_tir::ScheduleObjpure virtual
GetProducers(const SBlockRV &block_rv)=0tvm::s_tir::ScheduleObjpure virtual
GetSBlock(const ffi::String &name, const ffi::Optional< ffi::String > &func_name=std::nullopt)=0tvm::s_tir::ScheduleObjpure virtual
GetSRef(const SBlockRV &block_rv) const =0tvm::s_tir::ScheduleObjpure virtual
GetSRef(const LoopRV &loop_rv) const =0tvm::s_tir::ScheduleObjpure virtual
GetSRef(const StmtNode *stmt) consttvm::s_tir::ScheduleObjvirtual
GetSRef(const Stmt &stmt) consttvm::s_tir::ScheduleObjinline
HasBlock(const SBlockRV &block_rv) const =0tvm::s_tir::ScheduleObjpure virtual
LoopPartition(const LoopRV &loop_rv, const ffi::Array< ffi::Optional< ExprRV > > &factors, bool preserve_unit_iters=true)=0tvm::s_tir::ScheduleObjpure virtual
Merge(const ffi::Array< LoopRV > &loop_rvs)=0tvm::s_tir::ScheduleObjpure virtual
mod() consttvm::s_tir::ScheduleObjinlinevirtual
PadEinsum(const SBlockRV &block_rv, const ffi::Array< int64_t > &padding)=0tvm::s_tir::ScheduleObjpure virtual
Parallel(const LoopRV &loop_rv)=0tvm::s_tir::ScheduleObjpure virtual
ReadAt(const LoopRV &loop_rv, const SBlockRV &block_rv, int read_tensor_index, const ffi::String &storage_scope)=0tvm::s_tir::ScheduleObjpure virtual
ReIndex(const SBlockRV &block_rv, int tensor_index, TensorIndexType tensor_index_type)=0tvm::s_tir::ScheduleObjpure virtual
ReindexCacheRead(const SBlockRV &block_rv, int read_tensor_index, const ffi::String &storage_scope, const IndexMap &index_map)=0tvm::s_tir::ScheduleObjpure virtual
ReindexCacheWrite(const SBlockRV &block_rv, int write_tensor_index, const ffi::String &storage_scope, const IndexMap &index_map)=0tvm::s_tir::ScheduleObjpure virtual
RemoveRV(const SBlockRV &block_rv)=0tvm::s_tir::ScheduleObjpure virtual
RemoveRV(const LoopRV &loop_rv)=0tvm::s_tir::ScheduleObjpure virtual
RemoveRV(const ExprRV &expr_rv)=0tvm::s_tir::ScheduleObjpure virtual
Reorder(const ffi::Array< LoopRV > &ordered_loop_rvs)=0tvm::s_tir::ScheduleObjpure virtual
ReorderBlockIterVar(const SBlockRV &block_rv, const ffi::Array< int64_t > new_order)=0tvm::s_tir::ScheduleObjpure virtual
ReverseComputeAt(const SBlockRV &block_rv, const LoopRV &loop_rv, bool preserve_unit_loops, int index=-1)=0tvm::s_tir::ScheduleObjpure virtual
ReverseComputeInline(const SBlockRV &block)=0tvm::s_tir::ScheduleObjpure virtual
RFactor(const LoopRV &loop_rv, int factor_axis)=0tvm::s_tir::ScheduleObjpure virtual
RollingBuffer(const SBlockRV &block_rv, int write_tensor_index)=0tvm::s_tir::ScheduleObjpure virtual
SampleCategorical(const ffi::Array< int64_t > &candidates, const ffi::Array< FloatImm > &probs, ffi::Optional< int64_t > decision=std::nullopt)=0tvm::s_tir::ScheduleObjpure virtual
SampleComputeLocation(const SBlockRV &block_rv, ffi::Optional< int64_t > decision=std::nullopt)=0tvm::s_tir::ScheduleObjpure virtual
SamplePartitionedTile(const LoopRV &loop_rv, int n, int partition_pos, int innerpart_factor, ffi::Optional< ffi::Array< int64_t > > decision=std::nullopt)=0tvm::s_tir::ScheduleObjpure virtual
SamplePerfectTile(const LoopRV &loop_rv, int n, int max_innermost_factor, ffi::Optional< ffi::Array< int64_t > > decision=std::nullopt)=0tvm::s_tir::ScheduleObjpure virtual
Scheduletvm::s_tir::ScheduleObjfriend
Seed(LinearCongruentialEngine::TRandState seed)=0tvm::s_tir::ScheduleObjpure virtual
SetScope(const SBlockRV &block_rv, int tensor_index, const ffi::String &storage_scope)=0tvm::s_tir::ScheduleObjpure virtual
Split(const LoopRV &loop_rv, const ffi::Array< ffi::Optional< ExprRV > > &factors, bool preserve_unit_iters=true, bool disable_predication=false)=0tvm::s_tir::ScheduleObjpure virtual
state() const =0tvm::s_tir::ScheduleObjpure virtual
StorageAlign(const SBlockRV &block_rv, int tensor_index, int axis, int factor, int offset)=0tvm::s_tir::ScheduleObjpure virtual
Tensorize(const LoopRV &loop_rv, const ffi::String &intrin, bool preserve_unit_iters=true)=0tvm::s_tir::ScheduleObjpure virtual
Tensorize(const SBlockRV &block_rv, const ffi::String &intrin, bool preserve_unit_iters=true)=0tvm::s_tir::ScheduleObjpure virtual
trace() const =0tvm::s_tir::ScheduleObjpure virtual
TransformBlockLayout(const SBlockRV &block_rv, const IndexMap &index_map)=0tvm::s_tir::ScheduleObjpure virtual
TransformLayout(const SBlockRV &block_rv, int tensor_index, TensorIndexType tensor_index_type, const IndexMap &index_map, const ffi::Optional< IndexMap > &pad_value=std::nullopt, bool assume_injective_transform=false)=0tvm::s_tir::ScheduleObjpure virtual
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("s_tir.Schedule", ScheduleObj, ffi::Object)tvm::s_tir::ScheduleObj
Unannotate(const LoopRV &loop_rv, const ffi::String &ann_key)=0tvm::s_tir::ScheduleObjpure virtual
Unannotate(const SBlockRV &block_rv, const ffi::String &ann_key)=0tvm::s_tir::ScheduleObjpure virtual
Unroll(const LoopRV &loop_rv)=0tvm::s_tir::ScheduleObjpure virtual
UnsafeHideTensorAccess(const SBlockRV &block_rv, const ffi::String &tensor_type, const ffi::Array< IntImm > &tensor_index_array)=0tvm::s_tir::ScheduleObjpure virtual
UnsafeSetDType(const SBlockRV &block_rv, int tensor_index, const ffi::String &dtype)=0tvm::s_tir::ScheduleObjpure virtual
Vectorize(const LoopRV &loop_rv)=0tvm::s_tir::ScheduleObjpure virtual
WorkOn(const ffi::String &func_name)=0tvm::s_tir::ScheduleObjpure virtual
WriteAt(const LoopRV &loop_rv, const SBlockRV &block_rv, int write_tensor_index, const ffi::String &storage_scope)=0tvm::s_tir::ScheduleObjpure virtual
~ScheduleObj()=defaulttvm::s_tir::ScheduleObjvirtual