tvm
|
This is the complete list of members for tvm::tir::ScheduleNode, including all inherited members.
_type_key | tvm::tir::ScheduleNode | static |
AddUnitLoop(const BlockRV &block_rv)=0 | tvm::tir::ScheduleNode | pure virtual |
AddUnitLoop(const LoopRV &loop_rv)=0 | tvm::tir::ScheduleNode | pure virtual |
Annotate(const LoopRV &loop_rv, const String &ann_key, const Any &ann_val)=0 | tvm::tir::ScheduleNode | pure virtual |
Annotate(const BlockRV &block_rv, const String &ann_key, const Any &ann_val)=0 | tvm::tir::ScheduleNode | pure virtual |
AnnotateBufferAccess(const BlockRV &block_rv, int buffer_index, BufferIndexType buffer_index_type, const IndexMap &index_map)=0 | tvm::tir::ScheduleNode | pure virtual |
Bind(const LoopRV &loop_rv, const String &thread_axis)=0 | tvm::tir::ScheduleNode | pure virtual |
Blockize(const LoopRV &loop_rv, bool preserve_unit_iters=true)=0 | tvm::tir::ScheduleNode | pure virtual |
Blockize(const Array< BlockRV > &blocks, bool preserve_unit_iters=true)=0 | tvm::tir::ScheduleNode | pure virtual |
CacheIndex(const BlockRV &block_rv, const String &storage_scope, int cse_thresh)=0 | tvm::tir::ScheduleNode | pure virtual |
CacheInplace(const BlockRV &block_rv, int read_buffer_index, const String &storage_scope)=0 | tvm::tir::ScheduleNode | pure virtual |
CacheRead(const BlockRV &block_rv, int read_buffer_index, const String &storage_scope, const Array< BlockRV > consumer_blocks={})=0 | tvm::tir::ScheduleNode | pure virtual |
CacheWrite(const BlockRV &block_rv, int write_buffer_index, const String &storage_scope, const Array< BlockRV > consumer_blocks={})=0 | tvm::tir::ScheduleNode | pure virtual |
ComputeAt(const BlockRV &block_rv, const LoopRV &loop_rv, bool preserve_unit_loops, int index=-1)=0 | tvm::tir::ScheduleNode | pure virtual |
ComputeInline(const BlockRV &block)=0 | tvm::tir::ScheduleNode | pure virtual |
Copy()=0 | tvm::tir::ScheduleNode | pure virtual |
DecomposePadding(const BlockRV &block_rv, const LoopRV &loop_rv)=0 | tvm::tir::ScheduleNode | pure virtual |
DecomposeReduction(const BlockRV &block_rv, const LoopRV &loop_rv)=0 | tvm::tir::ScheduleNode | pure virtual |
EnterPostproc()=0 | tvm::tir::ScheduleNode | pure virtual |
ForkSeed()=0 | tvm::tir::ScheduleNode | pure virtual |
func_working_on() const =0 | tvm::tir::ScheduleNode | pure virtual |
Fuse(const Array< LoopRV > &loop_rvs, bool preserve_unit_iters=true)=0 | tvm::tir::ScheduleNode | pure virtual |
Get(const BlockRV &block_rv) const =0 | tvm::tir::ScheduleNode | pure virtual |
Get(const LoopRV &loop_rv) const =0 | tvm::tir::ScheduleNode | pure virtual |
Get(const ExprRV &expr_rv) const =0 | tvm::tir::ScheduleNode | pure virtual |
GetBlock(const String &name, const Optional< String > &func_name=std::nullopt)=0 | tvm::tir::ScheduleNode | pure virtual |
GetChildBlocks(const BlockRV &block_rv)=0 | tvm::tir::ScheduleNode | pure virtual |
GetChildBlocks(const LoopRV &loop_rv)=0 | tvm::tir::ScheduleNode | pure virtual |
GetConsumers(const BlockRV &block_rv)=0 | tvm::tir::ScheduleNode | pure virtual |
GetLoops(const BlockRV &block_rv)=0 | tvm::tir::ScheduleNode | pure virtual |
GetOutputBlocks(const BlockRV &scope_block_rv)=0 | tvm::tir::ScheduleNode | pure virtual |
GetProducers(const BlockRV &block_rv)=0 | tvm::tir::ScheduleNode | pure virtual |
GetSRef(const BlockRV &block_rv) const =0 | tvm::tir::ScheduleNode | pure virtual |
GetSRef(const LoopRV &loop_rv) const =0 | tvm::tir::ScheduleNode | pure virtual |
GetSRef(const StmtNode *stmt) const | tvm::tir::ScheduleNode | virtual |
GetSRef(const Stmt &stmt) const | tvm::tir::ScheduleNode | inline |
HasBlock(const BlockRV &block_rv) const =0 | tvm::tir::ScheduleNode | pure virtual |
LoopPartition(const LoopRV &loop_rv, const Array< Optional< ExprRV >> &factors, bool preserve_unit_iters=true)=0 | tvm::tir::ScheduleNode | pure virtual |
Merge(const Array< LoopRV > &loop_rvs)=0 | tvm::tir::ScheduleNode | pure virtual |
mod() const | tvm::tir::ScheduleNode | inlinevirtual |
PadEinsum(const BlockRV &block_rv, const Array< Integer > &padding)=0 | tvm::tir::ScheduleNode | pure virtual |
Parallel(const LoopRV &loop_rv)=0 | tvm::tir::ScheduleNode | pure virtual |
ReadAt(const LoopRV &loop_rv, const BlockRV &block_rv, int read_buffer_index, const String &storage_scope)=0 | tvm::tir::ScheduleNode | pure virtual |
ReIndex(const BlockRV &block_rv, int buffer_index, BufferIndexType buffer_index_type)=0 | tvm::tir::ScheduleNode | pure virtual |
ReindexCacheRead(const BlockRV &block_rv, int read_buffer_index, const String &storage_scope, const IndexMap &index_map)=0 | tvm::tir::ScheduleNode | pure virtual |
ReindexCacheWrite(const BlockRV &block_rv, int write_buffer_index, const String &storage_scope, const IndexMap &index_map)=0 | tvm::tir::ScheduleNode | pure virtual |
RemoveRV(const BlockRV &block_rv)=0 | tvm::tir::ScheduleNode | pure virtual |
RemoveRV(const LoopRV &loop_rv)=0 | tvm::tir::ScheduleNode | pure virtual |
RemoveRV(const ExprRV &expr_rv)=0 | tvm::tir::ScheduleNode | pure virtual |
Reorder(const Array< LoopRV > &ordered_loop_rvs)=0 | tvm::tir::ScheduleNode | pure virtual |
ReorderBlockIterVar(const BlockRV &block_rv, const Array< Integer > new_order)=0 | tvm::tir::ScheduleNode | pure virtual |
ReverseComputeAt(const BlockRV &block_rv, const LoopRV &loop_rv, bool preserve_unit_loops, int index=-1)=0 | tvm::tir::ScheduleNode | pure virtual |
ReverseComputeInline(const BlockRV &block)=0 | tvm::tir::ScheduleNode | pure virtual |
RFactor(const LoopRV &loop_rv, int factor_axis)=0 | tvm::tir::ScheduleNode | pure virtual |
RollingBuffer(const BlockRV &block_rv, int write_buffer_index)=0 | tvm::tir::ScheduleNode | pure virtual |
SampleCategorical(const Array< Integer > &candidates, const Array< FloatImm > &probs, Optional< Integer > decision=std::nullopt)=0 | tvm::tir::ScheduleNode | pure virtual |
SampleComputeLocation(const BlockRV &block_rv, Optional< Integer > decision=std::nullopt)=0 | tvm::tir::ScheduleNode | pure virtual |
SamplePartitionedTile(const LoopRV &loop_rv, int n, int partition_pos, int innerpart_factor, Optional< Array< Integer >> decision=std::nullopt)=0 | tvm::tir::ScheduleNode | pure virtual |
SamplePerfectTile(const LoopRV &loop_rv, int n, int max_innermost_factor, Optional< Array< Integer >> decision=std::nullopt)=0 | tvm::tir::ScheduleNode | pure virtual |
Schedule class | tvm::tir::ScheduleNode | friend |
Seed(support::LinearCongruentialEngine::TRandState seed)=0 | tvm::tir::ScheduleNode | pure virtual |
SetAxisSeparator(const BlockRV &block_rv, int buffer_index, BufferIndexType buffer_index_type, const Array< IntImm > &axis_separators)=0 | tvm::tir::ScheduleNode | pure virtual |
SetScope(const BlockRV &block_rv, int buffer_index, const String &storage_scope)=0 | tvm::tir::ScheduleNode | pure virtual |
Split(const LoopRV &loop_rv, const Array< Optional< ExprRV >> &factors, bool preserve_unit_iters=true, bool disable_predication=false)=0 | tvm::tir::ScheduleNode | pure virtual |
state() const =0 | tvm::tir::ScheduleNode | pure virtual |
StorageAlign(const BlockRV &block_rv, int buffer_index, int axis, int factor, int offset)=0 | tvm::tir::ScheduleNode | pure virtual |
Tensorize(const LoopRV &loop_rv, const String &intrin, bool preserve_unit_iters=true)=0 | tvm::tir::ScheduleNode | pure virtual |
Tensorize(const BlockRV &block_rv, const String &intrin, bool preserve_unit_iters=true)=0 | tvm::tir::ScheduleNode | pure virtual |
trace() const =0 | tvm::tir::ScheduleNode | pure virtual |
TransformBlockLayout(const BlockRV &block_rv, const IndexMap &index_map)=0 | tvm::tir::ScheduleNode | pure virtual |
TransformLayout(const BlockRV &block_rv, int buffer_index, BufferIndexType buffer_index_type, const IndexMap &index_map, const Optional< IndexMap > &pad_value=std::nullopt, bool assume_injective_transform=false)=0 | tvm::tir::ScheduleNode | pure virtual |
TVM_DECLARE_FINAL_OBJECT_INFO(ScheduleNode, runtime::Object) | tvm::tir::ScheduleNode | |
Unannotate(const LoopRV &loop_rv, const String &ann_key)=0 | tvm::tir::ScheduleNode | pure virtual |
Unannotate(const BlockRV &block_rv, const String &ann_key)=0 | tvm::tir::ScheduleNode | pure virtual |
Unroll(const LoopRV &loop_rv)=0 | tvm::tir::ScheduleNode | pure virtual |
UnsafeHideBufferAccess(const BlockRV &block_rv, const String &buf_type, const Array< IntImm > &buf_index_array)=0 | tvm::tir::ScheduleNode | pure virtual |
UnsafeSetDType(const BlockRV &block_rv, int buffer_index, const String &dtype)=0 | tvm::tir::ScheduleNode | pure virtual |
Vectorize(const LoopRV &loop_rv)=0 | tvm::tir::ScheduleNode | pure virtual |
WorkOn(const String &func_name)=0 | tvm::tir::ScheduleNode | pure virtual |
WriteAt(const LoopRV &loop_rv, const BlockRV &block_rv, int write_buffer_index, const String &storage_scope)=0 | tvm::tir::ScheduleNode | pure virtual |
~ScheduleNode()=default | tvm::tir::ScheduleNode | virtual |