Python API#
- tvm.runtime
DataType
DataTypeCode
Device
Module
Object
ObjectGeneric
ObjectPath
ObjectPathPair
PackedFunc
Report
Scriptable
ShapeTuple
String
cl()
const()
convert()
convert_to_object()
cpu()
cuda()
device()
enabled()
ext_dev()
gpu()
load_module()
load_param_dict()
load_param_dict_from_file()
load_static_library()
metal()
mtl()
num_threads()
opencl()
rocm()
save_param_dict()
save_param_dict_to_file()
system_lib()
vpi()
vulkan()
- tvm.runtime.ndarray
- tvm.error
- tvm.ir
Array
Attrs
BaseExpr
BaseFunc
CallingConv
ConstantMemoryPools
ConstantPoolInfo
Constructor
DictAttrs
DummyGlobalInfo
EnvFunc
FuncType
GlobalInfo
GlobalTypeVar
GlobalVar
IRModule
IncompleteType
Map
Node
Op
PointerType
PoolInfo
PoolInfoProperties
PrimExpr
PrimType
Range
RelayExpr
RelayRefType
SequentialSpan
SourceName
Span
TensorAffineType
TensorType
TupleAffineType
TupleType
Type
TypeCall
TypeConstraint
TypeData
TypeKind
TypeRelation
TypeVar
VDevice
WorkspaceMemoryPools
WorkspacePoolInfo
assert_structural_equal()
load_json()
make_node()
register_intrin_lowering()
register_op_attr()
save_json()
structural_equal()
structural_hash()
- tvm.instrument
- tvm.transform
- tvm.target
- tvm.tir
ScheduleError
Add
Allocate
AllocateConst
And
Any
AssertStmt
AttrStmt
BijectiveLayout
Block
BlockDependenceInfo
BlockRealize
BlockScope
Broadcast
Buffer
BufferLoad
BufferRealize
BufferRegion
BufferStore
Call
CallEffectKind
Cast
CommReducer
DataProducer
DeclBuffer
Div
EQ
Evaluate
FloatImm
FloorDiv
FloorMod
For
ForKind
GE
GT
IfThenElse
IndexMap
IntImm
IterVar
LE
LT
Layout
Let
LetStmt
MatchBufferRegion
Max
Min
Mod
Mul
NE
Not
Or
Prefetch
PrimFunc
ProducerLoad
ProducerRealize
ProducerStore
Ramp
Reduce
Schedule
ScheduleState
Select
SeqStmt
Shuffle
SizeVar
Stmt
StmtSRef
StringImm
Sub
TensorIntrin
Var
While
TVMBackendAllocWorkspace()
TVMBackendFreeWorkspace()
abs()
acos()
acosh()
add()
address_of()
all()
any()
asin()
asinh()
assume()
atan()
atan2()
atanh()
bijective_layout()
bitwise_and()
bitwise_not()
bitwise_or()
bitwise_xor()
call_cpacked()
call_cpacked_lowered()
call_extern()
call_intrin()
call_llvm_intrin()
call_llvm_pure_intrin()
call_packed()
call_packed_lowered()
call_pure_extern()
call_tir()
ceil()
ceildiv()
clz()
comm_reducer()
copysign()
cos()
cosh()
create_barriers()
decl_buffer()
div()
end_profile_intrinsic()
erf()
exp()
exp10()
exp2()
floor()
floordiv()
floormod()
fmod()
get_active_lane_mask()
hypot()
if_then_else()
indexdiv()
indexmod()
infinity()
isfinite()
isinf()
isnan()
isnullptr()
layout()
ldexp()
likely()
log()
log10()
log1p()
log2()
lookup_param()
max()
max_value()
min()
min_value()
mma_fill()
mma_store()
multiply()
nearbyint()
nextafter()
popcount()
pow()
power()
ptx_arrive_barrier()
ptx_arrive_barrier_expect_tx()
ptx_commit_group()
ptx_cp_async()
ptx_cp_async_barrier()
ptx_cp_async_bulk()
ptx_init_barrier_thread_count()
ptx_ldmatrix()
ptx_mma()
ptx_mma_sp()
ptx_wait_barrier()
ptx_wait_group()
q_multiply_shift()
q_multiply_shift_per_axis()
reinterpret()
ret()
round()
rsqrt()
shift_left()
shift_right()
sigmoid()
sin()
sinh()
sqrt()
start_profile_intrinsic()
stmt_list()
stmt_seq()
subtract()
sum()
tan()
tanh()
trace()
trunc()
truncdiv()
truncmod()
tvm_access_ptr()
tvm_bmma_sync()
tvm_check_return()
tvm_fill_fragment()
tvm_load_matrix_sync()
tvm_mma_sync()
tvm_stack_alloca()
tvm_stack_make_array()
tvm_stack_make_shape()
tvm_store_matrix_sync()
tvm_struct_get()
tvm_struct_set()
tvm_thread_allreduce()
tvm_throw_last_error()
tvm_tuple()
type_annotation()
undef()
vectorcombine()
vectorhigh()
vectorlow()
vscale()
- tvm.tir.transform
HoistedConditionals
HoistedLetBindings
PrimFuncPass
AnnotateDeviceRegions()
AnnotateEntryFunc()
Apply()
ApplyLayoutTransforms()
BF16ComputeLegalize()
BF16StorageLegalize()
BindTarget()
CoProcSync()
CombineContextCall()
CommonSubexprElimTIR()
CompactBufferAllocation()
ConvertBlocksToOpaque()
ConvertForLoopsToSerial()
ConvertSSA()
DecorateDeviceScope()
DefaultGPUSchedule()
ExtractPrimFuncConstants()
FP8ComputeLegalize()
FP8StorageLegalize()
Filter()
FlattenBuffer()
ForceNarrowIndexToInt32()
HoistExpression()
HoistIfThenElse()
InferFragment()
InjectCopyIntrin()
InjectDoubleBuffer()
InjectPTXAsyncCopy()
InjectPermutedLayout()
InjectPrefetch()
InjectRollingBuffer()
InjectSoftwarePipeline()
InjectVirtualThread()
InlinePrivateFunctions()
InstallDebugSpans()
InstrumentBoundCheckers()
InstrumentProfileIntrinsics()
LegalizePackedCalls()
LiftAttrScope()
LiftThreadBinding()
LoopPartition()
LowerAutoCopy()
LowerCrossThreadReduction()
LowerCustomDatatypes()
LowerDeviceKernelLaunch()
LowerDeviceStorageAccessInfo()
LowerInitBlock()
LowerIntrin()
LowerMatchBuffer()
LowerOpaqueBlock()
LowerTVMBuiltin()
LowerThreadAllreduce()
LowerWarpMemory()
MakePackedAPI()
MakeUnpackedAPI()
ManifestSharedMemoryLocalStage()
MergeSharedMemoryAllocations()
NarrowDataType()
PlanAndUpdateBufferAllocationLocation()
PointerValueTypeRewrite()
ReduceBranchingThroughOvercompute()
RemoveAssume()
RemoveNoOp()
RemoveStoreUndef()
RemoveWeightLayoutRewriteBlock()
RenormalizeSplitPattern()
RewriteUnsafeSelect()
Simplify()
SkipAssert()
SplitHostDevice()
StorageFlatten()
StorageRewrite()
TextureFlatten()
ThreadSync()
TransformMmaBufferLayout()
UnifyThreadBinding()
UnrollLoop()
VectorizeLoop()
VerifyMemory()
VerifyVTCMLimit()
prim_func_pass()
- tvm.tir.analysis
- tvm.tir.stmt_functor
- tvm.te
ComputeOp
ExternOp
HybridOp
PlaceholderOp
ScanOp
Schedule
SpecializedCondition
Stage
Tensor
TensorComputeOp
TensorSlice
abs()
acos()
acosh()
add()
all()
any()
asin()
asinh()
atan()
atanh()
ceil()
comm_reducer()
compute()
const()
cos()
cosh()
create_prim_func()
create_schedule()
decl_tensor_intrin()
div()
erf()
exp()
extern()
extern_primfunc()
floor()
floordiv()
floormod()
fmod()
gradient()
if_then_else()
indexdiv()
indexmod()
isfinite()
isinf()
isnan()
log()
log10()
log2()
max()
max_value()
min()
min_value()
multiply()
nearbyint()
placeholder()
popcount()
power()
reduce_axis()
round()
rsqrt()
scan()
sigmoid()
sin()
sinh()
size_var()
sqrt()
subtract()
sum()
tag_scope()
tan()
tanh()
thread_axis()
trace()
trunc()
truncdiv()
truncmod()
var()
- tvm.te.hybrid
- tvm.driver
- tvm.relay
- tvm.relay.frontend
- tvm.relay.nn
Constant
Expr
adaptive_avg_pool1d()
adaptive_avg_pool2d()
adaptive_avg_pool3d()
adaptive_max_pool1d()
adaptive_max_pool2d()
adaptive_max_pool3d()
avg_pool1d()
avg_pool2d()
avg_pool2d_grad()
avg_pool3d()
batch_flatten()
batch_matmul()
batch_norm()
batch_to_space_nd()
bias_add()
bitpack()
bitserial_conv2d()
bitserial_dense()
const()
contrib_conv2d_gemm_weight_transform()
contrib_conv2d_gemm_without_weight_transform()
contrib_conv2d_nchwc()
contrib_conv2d_winograd_nnpack_weight_transform()
contrib_conv2d_winograd_weight_transform()
contrib_conv2d_winograd_without_weight_transform()
contrib_conv3d_winograd_weight_transform()
contrib_conv3d_winograd_without_weight_transform()
contrib_dense_pack()
contrib_depthwise_conv2d_nchwc()
conv1d()
conv1d_transpose()
conv2d()
conv2d_backward_weight()
conv2d_transpose()
conv3d()
conv3d_transpose()
correlation()
cross_entropy()
cross_entropy_with_logits()
deformable_conv2d()
dense()
depth_to_space()
dilate()
dropout()
dropout_raw()
fast_softmax()
fifo_buffer()
get_pad_tuple1d()
get_pad_tuple2d()
get_pad_tuple3d()
global_avg_pool1d()
global_avg_pool2d()
global_avg_pool3d()
global_max_pool1d()
global_max_pool2d()
global_max_pool3d()
group_norm()
instance_norm()
l2_normalize()
layer_norm()
leaky_relu()
log_softmax()
lrn()
matmul()
max_pool1d()
max_pool2d()
max_pool2d_grad()
max_pool3d()
mirror_pad()
nll_loss()
pad()
prelu()
relu()
softmax()
space_to_batch_nd()
space_to_depth()
sparse_add()
sparse_dense()
sparse_transpose()
upsampling()
upsampling3d()
- tvm.relay.vision
- tvm.relay.image
- tvm.relay.transform
ChangeBatch
FlexibleShapeDispatch
FunctionPass
LayoutConfig
AlterOpLayout()
AnnotateSpans()
AnnotateTarget()
BackwardFoldScaleAxis()
BatchingOps()
CanonicalizeCast()
CanonicalizeOps()
CapturePostDfsIndexInSpans()
CollagePartition()
CombineParallelBatchMatmul()
CombineParallelConv2D()
CombineParallelDense()
Conv2dToSparse()
Conv2dToSparse2()
ConvertLayout()
DeadCodeElimination()
Defunctionalization()
DefuseOps()
DenseToSparse()
DivToMul()
DynamicToStatic()
EliminateCommonSubexpr()
EtaExpand()
FakeQuantizationToInteger()
FastMath()
FirstOrderGradient()
FlattenAtrousConv()
FoldConstant()
FoldConstantExpr()
FoldExplicitPadding()
FoldScaleAxis()
ForwardFoldScaleAxis()
FuseOps()
InferType()
InferTypeLocal()
Inline()
InlineCompilerFunctionsBoundTo()
LambdaLift()
LazyGradientInit()
Legalize()
ManifestLifetimes()
MarkCompilerFunctionsAsExtern()
MergeCompilerRegions()
MergeComposite()
OutlineCompilerFunctionsWithExistingGlobalSymbols()
PartialEvaluate()
PartitionGraph()
PlanDevices()
RemoveUnusedFunctions()
SimplifyExpr()
SimplifyFCTranspose()
SimplifyInference()
SplitArgs()
ToANormalForm()
ToANormalFormExpr()
ToBasicBlockNormalForm()
ToCPS()
ToGraphNormalForm()
ToMixedPrecision()
build_config()
function_pass()
gradient()
recast()
to_cps()
un_cps()
- tvm.relay.analysis
AnnotatedRegionSet
CallGraph
Feature
all_dtypes()
all_type_vars()
all_vars()
bound_type_vars()
bound_vars()
check_basic_block_normal_form()
check_constant()
check_kind()
count_layers()
cpu()
detect_feature()
extract_fused_functions()
extract_intermdeiate_expr()
free_type_vars()
free_vars()
get_calibration_data()
get_total_mac_number()
list_fake_quantized_op_freqs()
list_op_freqs()
post_order_visit()
search_fc_transpose()
unmatched_cases()
well_formed()
- tvm.relay.backend
- tvm.relay.dataflow_pattern
AltPattern
AttrPattern
CallPattern
ConstantPattern
DFPattern
DFPatternCallback
DataTypePattern
DominatorPattern
Expr
ExprPattern
FunctionPattern
IfPattern
LetPattern
ShapePattern
TupleGetItemPattern
TuplePattern
TypePattern
VarPattern
WildcardPattern
_DFPatternCallback
astext()
dominates()
get()
has_attr()
has_dtype()
has_shape()
has_type()
is_constant()
is_expr()
is_if()
is_let()
is_op()
is_tuple()
is_tuple_get_item()
is_var()
make_node()
match()
partition()
pretty_print()
register_df_node()
rewrite()
wildcard()
- tvm.relay.testing
Prelude
check_grad()
count()
count_ops()
create_workload()
enabled_targets()
gradient()
make_nat_expr()
make_nat_value()
run_as_python()
to_python()
get_net()
get_workload()
get_net()
get_workload()
residual_unit()
resnet()
deconv2d()
deconv2d_bn_relu()
get_net()
get_workload()
conv_block()
get_workload()
mobile_net()
separable_conv_block()
get_net()
get_workload()
lstm_cell()
get_net()
get_workload()
get_net()
get_workload()
get_classifier()
get_feature()
get_net()
get_workload()
_make_dense_block()
_make_dense_layer()
_make_dense_net()
_make_transition()
get_workload()
- tvm.autotvm
- tvm.auto_scheduler
ApplyHistoryBest
ApplyHistoryBestOrSample
ComputeDAG
DispatchContext
EmptyPolicy
HardwareParams
LayoutRewriteOption
LocalBuilder
LocalRPCMeasureContext
LocalRunner
MeasureInput
MeasureResult
PreloadCustomSketchRule
PreloadMeasuredStates
RPCRunner
RandomModel
RecordReader
RecordToFile
SearchTask
SketchPolicy
TaskScheduler
TuningOptions
XGBModel
auto_schedule()
create_task()
extract_tasks()
get_shape_from_rewritten_layout()
is_auto_scheduler_enabled()
load_best_record()
load_records()
make_workload_key()
register_task_input_check_func()
register_workload()
remove_index_check()
rewrite_compute_body()
rewrite_tensor_shape()
save_records()
- tvm.meta_schedule
- tvm.rpc
- tvm.micro
SessionTerminatedError
UnsupportedInModelLibraryFormatError
AutoTvmModuleLoader
GeneratedProject
Session
TemplateProject
TransportLogger
autotvm_build_func()
copy_crt_config_header()
create_local_aot_executor()
create_local_debug_executor()
create_local_graph_executor()
export_model_library_format()
generate_project()
get_microtvm_template_projects()
get_standalone_crt_dir()
- tvm.contrib
- tvm.contrib.cblas
- tvm.contrib.clang
- tvm.contrib.cc
- tvm.contrib.cublas
- tvm.contrib.dlpack
- tvm.contrib.emcc
- tvm.contrib.miopen
- tvm.contrib.mxnet
- tvm.contrib.ndk
- tvm.contrib.nnpack
- tvm.contrib.nvcc
- tvm.contrib.pickle_memoize
- tvm.contrib.random
- tvm.contrib.relay_viz
- tvm.contrib.rocblas
- tvm.contrib.rocm
- tvm.contrib.sparse
- tvm.contrib.spirv
- tvm.contrib.tar
- tvm.contrib.utils
- tvm.contrib.xcode
- tvm.contrib.graph_executor
- tvm.topi
- vta