schedule.h File Reference

Define a schedule. More...

#include <tvm/support/with.h>
#include <tvm/te/tensor.h>
#include <tvm/te/tensor_intrin.h>
#include <tvm/tir/expr.h>
#include <tvm/tir/index_map.h>
#include <string>
#include <unordered_map>

Include dependency graph for schedule.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	tvm::te::Stage
	Stage, contains scheduling for a stage of computation. More...
class	tvm::te::Schedule
	Global schedule container For operations and all the operations they depend on. The schedule per Operation is named as stage. More...
class	tvm::te::ScheduleContext
	Context helper to collect debug information of Schedule. More...
class	tvm::te::IterVarRelation
	The schedule relation between IterVars can be Split, Fuse. More...
class	tvm::te::IterVarAttr
	Additional scheduable attributes about IterVar. More...
class	tvm::te::StageNode
	represents a stage. More...
class	tvm::te::ScheduleNode
	node container for schedule More...
class	tvm::te::IterVarAttrNode
	node container for IterVar attr More...
class	tvm::te::IterVarRelationNode
	base node of iteration var More...
class	tvm::te::SplitNode
	Split the parent domain into product of outer and iter. More...
class	tvm::te::Split
	Managed reference to SplitNode. More...
class	tvm::te::FuseNode
	Fuse two domains into one domain. More...
class	tvm::te::Fuse
	Managed reference to FuseNode. More...
class	tvm::te::RebaseNode
	Rebase the iteration to make min to be 0. This is useful to normalize the Schedule to make every leaf variable's min to be 0. More...
class	tvm::te::Rebase
	Managed reference to RebaseNode. More...
class	tvm::te::SingletonNode
	Singleton iterator [0, 1) More...
class	tvm::te::Singleton
	Managed reference to SingletonNode. More...
class	tvm::te::TransformNode
	Transform iterator according to some arbitrary expression. More...
class	tvm::te::Transform
class	tvm::te::SpecializedConditionNode
	Container for specialization conditions. More...
class	tvm::te::SpecializedCondition
	Specialized condition to enable op specialization. More...

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::te
	Tensor expression language DSL.

Enumerations
enum	tvm::te::AttachType : int {   tvm::te::kGroupRoot = 1 , tvm::te::kInline = 2 , tvm::te::kInlinedAlready = 3 , tvm::te::kScope = 4 ,   tvm::te::kScanUpdate = 5 }
	the attachment type More...

Functions
Schedule	tvm::te::create_schedule (Array< Operation > ops)
	Create a schedule for array of ops(and their dependencies). More...

Detailed Description

Define a schedule.