Community
Download
Blog
Docs
Conference
Github
ASF
Apache Homepage
License
Sponsorship
Security
Thanks
Events
ASF
Apache Homepage
License
Sponsorship
Security
Thanks
Events
tvm
Getting Started
Overview
Installing TVM
Quick Start
IRModule
How To
End-to-End Optimize Model
Customize Optimization
Optimize Large Language Model
Cross Compilation and RPC
Development Guides
Deep Dive
Design and Architecture
TensorIR
Relax
API Reference
Python API
Other APIs
About
Contributor Guide
Publications
Security Guide
Index
Index
Table of Contents
Docs
>
Index
Edit on GitHub
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
J
|
K
|
L
|
M
|
N
|
O
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
|
X
|
Z
A
abs() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
access_ptr() (tvm.tir.Buffer method)
acos() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
acosh() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
adaptive_avg_pool1d() (in module tvm.relax.op.nn)
adaptive_avg_pool2d() (in module tvm.relax.op.nn)
adaptive_avg_pool3d() (in module tvm.relax.op.nn)
adaptive_pool() (in module tvm.topi.nn)
adaptive_pool1d() (in module tvm.topi.nn)
adaptive_pool3d() (in module tvm.topi.nn)
AdaptivePool2DAttrs (class in tvm.relax.op.op_attrs)
Add (class in tvm.tir)
add() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
(in module tvm.topi.nn)
(tvm.relax.DataflowBlockRewrite method)
add_argument() (tvm.ir.Op method)
add_func() (tvm.relax.block_builder.BlockBuilder method)
add_unit_loop() (tvm.tir.schedule.Schedule method)
address_of() (in module tvm.tir)
AdjustMatmulOrder() (in module tvm.relax.transform)
adreno() (in module tvm.target)
affine_grid() (in module tvm.topi.image)
All (tvm.tir.transform.HoistedConditionals attribute)
(tvm.tir.transform.HoistedLetBindings attribute)
all() (in module tvm.te)
(in module tvm.tir)
all_global_vars() (in module tvm.relax.analysis)
all_vars() (in module tvm.relax.analysis)
allgather() (in module tvm.relax.op.ccl)
(tvm.runtime.disco.Session method)
alloc_storage() (in module tvm.relax.op.memory)
alloc_tensor() (in module tvm.relax.op.builtin)
(in module tvm.relax.op.memory)
Allocate (class in tvm.tir)
AllocateConst (class in tvm.tir)
AllocateWorkspace() (in module tvm.relax.transform)
allreduce() (in module tvm.relax.op.ccl)
(tvm.runtime.disco.Session method)
AlterOpImpl() (in module tvm.relax.transform)
And (class in tvm.tir)
annotate() (tvm.tir.schedule.Schedule method)
annotate_buffer_access() (tvm.tir.schedule.Schedule method)
annotate_sharding() (in module tvm.relax.op.distributed)
AnnotateDeviceRegions() (in module tvm.tir.transform)
AnnotateEntryFunc() (in module tvm.tir.transform)
AnnotateTIROpPattern() (in module tvm.relax.transform)
any() (in module tvm.te)
(in module tvm.tir)
append() (tvm.tir.schedule.Trace method)
Apply() (in module tvm.tir.transform)
apply() (tvm.dlight.ScheduleRule method)
(tvm.meta_schedule.MeasureCallback method)
(tvm.meta_schedule.Mutator method)
(tvm.meta_schedule.Postproc method)
(tvm.meta_schedule.ScheduleRule method)
apply_json_to_schedule() (tvm.tir.schedule.Trace static method)
apply_to_schedule() (tvm.tir.schedule.Trace method)
ApplyDefaultSchedule (class in tvm.dlight)
ApplyPassToFunction() (in module tvm.transform)
arange() (in module tvm.relax.op)
arch (tvm.target.Target property)
argmax() (in module tvm.relax.op)
ArgmaxArgminAttrs (class in tvm.relax.op.op_attrs)
argmin() (in module tvm.relax.op)
argsort() (in module tvm.relax.op)
ArgsortAttrs (class in tvm.relax.op.op_attrs)
arm_cpu() (in module tvm.target)
Array (class in tvm.ir)
array() (in module tvm.runtime.ndarray)
as_json() (tvm.tir.schedule.Trace method)
as_python() (tvm.relax.VMExecutable method)
(tvm.tir.schedule.Trace method)
as_text() (tvm.relax.VMExecutable method)
asin() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
asinh() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
asnumpy() (tvm.runtime.ndarray.NDArray method)
asobject() (tvm.te.TensorSlice method)
assert_op() (in module tvm.relax.op)
assert_pure_function() (in module tvm.tir.analysis.analysis)
assert_structural_equal() (in module tvm.ir)
AssertStmt (class in tvm.tir)
assume() (in module tvm.tir)
astype() (in module tvm.relax.op)
AstypeAttrs (class in tvm.relax.op.op_attrs)
atan() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
atan2() (in module tvm.tir)
atanh() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
AttachAttrLayoutFreeBuffers() (in module tvm.relax.transform)
AttachExternModules (class in tvm.relax.transform)
AttachGlobalSymbol() (in module tvm.relax.transform)
attention() (in module tvm.relax.op.nn)
attention_var_len() (in module tvm.relax.op.nn)
Attrs (class in tvm.ir)
attrs (tvm.ir.BaseFunc property)
(tvm.tir.schedule.Instruction attribute)
AttrStmt (class in tvm.tir)
avg_pool1d() (in module tvm.relax.op.nn)
avg_pool2d() (in module tvm.relax.op.nn)
avg_pool2d_backward() (in module tvm.relax.op.grad)
avg_pool3d() (in module tvm.relax.op.nn)
axis (tvm.te.Tensor property)
B
backward_index() (tvm.tir.BijectiveLayout method)
backward_shape() (tvm.tir.BijectiveLayout method)
BaseCheckResult (class in tvm.relax.analysis)
BaseExpr (class in tvm.ir)
BaseFunc (class in tvm.ir)
batch_matmul() (in module tvm.contrib.cblas)
(in module tvm.contrib.cublas)
(in module tvm.contrib.rocblas)
(in module tvm.topi.nn)
batch_norm() (in module tvm.relax.op.nn)
(in module tvm.topi.nn)
batch_to_space_nd() (in module tvm.topi.nn)
BatchNormAttrs (class in tvm.relax.op.op_attrs)
begin_scope() (tvm.relax.block_builder.BlockBuilder method)
BF16ComputeLegalize() (in module tvm.tir.transform)
BF16StorageLegalize() (in module tvm.tir.transform)
bifrost() (in module tvm.target)
bijective_layout() (in module tvm.tir)
BijectiveLayout (class in tvm.tir)
binarize_pack() (in module tvm.topi.nn)
binary_dense() (in module tvm.topi.nn)
bind() (tvm.tir.schedule.Schedule method)
bind_params() (tvm.relax.Function method)
bind_symbolic_vars() (tvm.relax.Function method)
Binding (class in tvm.relax)
BindingBlock (class in tvm.relax)
BindParams() (in module tvm.relax.transform)
BindSymbolicVars() (in module tvm.relax.transform)
BindTarget() (in module tvm.tir.transform)
bitpack() (in module tvm.topi.nn)
bitserial_conv2d_nchw() (in module tvm.topi.nn)
bitserial_conv2d_nhwc() (in module tvm.topi.nn)
bitserial_dense() (in module tvm.topi.nn)
bitwise_and() (in module tvm.relax.op)
(in module tvm.tir)
bitwise_not() (in module tvm.relax.op)
(in module tvm.tir)
bitwise_or() (in module tvm.relax.op)
(in module tvm.tir)
bitwise_xor() (in module tvm.relax.op)
(in module tvm.tir)
Block (class in tvm.tir)
BlockBuilder (class in tvm.relax.block_builder)
BlockDependenceInfo (class in tvm.tir)
BlockInfo (class in tvm.dlight)
blockize() (tvm.tir.schedule.Schedule method)
BlockRealize (class in tvm.tir)
BlockRV (class in tvm.tir.schedule)
BlockScope (class in tvm.tir.schedule)
BooleanExpression (tvm.tir.transform.HoistedConditionals attribute)
bound_vars() (in module tvm.relax.analysis)
Broadcast (class in tvm.tir)
broadcast() (tvm.runtime.disco.Session method)
broadcast_from_worker0() (in module tvm.relax.op.ccl)
(tvm.runtime.disco.Session method)
broadcast_to() (in module tvm.relax.op)
Buffer (class in tvm.tir)
BufferLoad (class in tvm.tir)
BufferRealize (class in tvm.tir)
BufferRegion (class in tvm.tir)
BufferStore (class in tvm.tir)
build() (in module tvm.relax)
(in module tvm.tir)
(tvm.meta_schedule.Builder method)
Builder (class in tvm.meta_schedule)
BundleModelParams() (in module tvm.relax.transform)
C
c() (tvm.relax.ExecBuilder method)
Cache (class in tvm.contrib.pickle_memoize)
cache (tvm.contrib.pickle_memoize.Cache property)
cache_index() (tvm.tir.schedule.Schedule method)
cache_inplace() (tvm.tir.schedule.Schedule method)
cache_read() (tvm.tir.schedule.Schedule method)
cache_write() (tvm.tir.schedule.Schedule method)
calculate_allocated_bytes() (in module tvm.tir.analysis.analysis)
Call (class in tvm.relax)
(class in tvm.tir)
call_builtin_with_ctx() (in module tvm.relax.op)
call_cpacked() (in module tvm.tir)
call_cpacked_lowered() (in module tvm.tir)
call_dps_packed() (in module tvm.relax)
(in module tvm.relax.op)
call_extern() (in module tvm.tir)
call_inplace_packed() (in module tvm.relax.op)
call_intrin() (in module tvm.tir)
call_llvm_intrin() (in module tvm.tir)
call_llvm_pure_intrin() (in module tvm.tir)
call_packed() (in module tvm.tir)
(tvm.runtime.disco.Session method)
call_packed_lowered() (in module tvm.tir)
call_pure_extern() (in module tvm.tir)
call_pure_packed() (in module tvm.relax)
(in module tvm.relax.op)
call_te() (tvm.relax.block_builder.BlockBuilder method)
call_te_with_grad() (tvm.relax.block_builder.BlockBuilder method)
call_tir() (in module tvm.relax)
(in module tvm.relax.op)
(in module tvm.tir)
call_tir_inplace() (in module tvm.relax)
(in module tvm.relax.op)
call_tir_local_view() (in module tvm.relax.op.distributed)
call_tir_with_grad() (in module tvm.relax)
(in module tvm.relax.op)
CallEffectKind (class in tvm.tir)
CallingConv (class in tvm.ir)
calls (tvm.runtime.profiling.Report attribute)
CallTIRRewrite() (in module tvm.relax.transform)
CallTIRWithGradAttrs (class in tvm.relax.op.op_attrs)
can_convert_multiply_to_intdiv() (in module tvm.topi.image)
can_decompose_padding() (tvm.tir.schedule.Schedule method)
canon_multi_target() (tvm.target.Target static method)
canon_multi_target_and_host() (tvm.target.Target static method)
canon_target() (tvm.target.Target static method)
canon_target_and_host() (tvm.target.Target static method)
canon_target_map_and_host() (tvm.target.Target static method)
CanonicalizeBindings() (in module tvm.relax.transform)
Cast (class in tvm.tir)
ceil() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
ceildiv() (in module tvm.tir)
checked_type (tvm.ir.RelaxExpr property)
cl() (in module tvm.runtime.ndarray)
(tvm.rpc.RPCSession method)
clip() (in module tvm.relax.op)
clone() (tvm.meta_schedule.Mutator method)
(tvm.meta_schedule.Postproc method)
(tvm.meta_schedule.ScheduleRule method)
(tvm.meta_schedule.SearchStrategy method)
(tvm.meta_schedule.SpaceGenerator method)
(tvm.meta_schedule.TuneContext method)
close() (tvm.rpc.TrackerSession method)
clz() (in module tvm.tir)
collapse_sum_like() (in module tvm.relax.op)
collapse_sum_to() (in module tvm.relax.op)
collect_non_negative_expressions() (in module tvm.relax.analysis)
CombineContextCall() (in module tvm.tir.transform)
CombineParallelMatmul() (in module tvm.relax.transform)
comm_reducer() (in module tvm.te)
(in module tvm.tir)
commit_tuning_record() (tvm.meta_schedule.Database method)
commit_workload() (tvm.meta_schedule.Database method)
CommonSubexprElimTIR() (in module tvm.tir.transform)
CommReducer (class in tvm.tir)
CompactBufferAllocation() (in module tvm.tir.transform)
compile() (in module tvm)
compile_coreml() (in module tvm.contrib.xcode)
compile_cuda() (in module tvm.contrib.nvcc)
compile_metal() (in module tvm.contrib.xcode)
computable_at_compile_time() (in module tvm.relax.analysis)
compute() (in module tvm.te)
compute_at() (tvm.tir.schedule.Schedule method)
compute_inline() (tvm.tir.schedule.Schedule method)
ComputeOp (class in tvm.te)
ComputePrimValue() (in module tvm.relax.transform)
concat() (in module tvm.relax.op)
ConcatAttrs (class in tvm.relax.op.op_attrs)
concatenate() (in module tvm.topi.nn)
connect() (in module tvm.rpc)
connect_tracker() (in module tvm.rpc)
const() (in module tvm.relax)
(in module tvm.te)
Constant (class in tvm.relax)
contains_impure_call() (in module tvm.relax.analysis)
conv() (in module tvm.topi.nn)
conv1d() (in module tvm.relax.op.nn)
(in module tvm.topi.nn)
conv1d_ncw() (in module tvm.topi.nn)
conv1d_nwc() (in module tvm.topi.nn)
conv1d_transpose() (in module tvm.relax.op.nn)
conv1d_transpose_ncw() (in module tvm.topi.nn)
conv2d() (in module tvm.relax.op.nn)
(in module tvm.topi.nn)
conv2d_forward() (in module tvm.contrib.miopen)
conv2d_hwcn() (in module tvm.topi.nn)
conv2d_nchw() (in module tvm.topi.nn)
conv2d_NCHWc() (in module tvm.topi.nn)
conv2d_NCHWc_int8() (in module tvm.topi.nn)
conv2d_nhwc() (in module tvm.topi.nn)
conv2d_transpose() (in module tvm.relax.op.nn)
conv2d_transpose_nchw() (in module tvm.topi.nn)
conv2d_transpose_nchw_preprocess() (in module tvm.topi.nn)
conv2d_winograd_nchw() (in module tvm.topi.nn)
conv2d_winograd_nchw_without_weight_transform() (in module tvm.topi.nn)
conv2d_winograd_nhwc() (in module tvm.topi.nn)
conv2d_winograd_nhwc_without_weight_transform() (in module tvm.topi.nn)
conv2d_winograd_weight_transform() (in module tvm.topi.nn)
Conv2DAttrs (class in tvm.relax.op.op_attrs)
Conv2DTransposeAttrs (class in tvm.relax.op.op_attrs)
conv3d() (in module tvm.relax.op.nn)
conv3d_ncdhw() (in module tvm.topi.nn)
conv3d_ndhwc() (in module tvm.topi.nn)
conv3d_transpose_ncdhw() (in module tvm.topi.nn)
conv3d_transpose_ncdhw_preprocess() (in module tvm.topi.nn)
conv3d_winograd_weight_transform() (in module tvm.topi.nn)
Conv3DAttrs (class in tvm.relax.op.op_attrs)
convert_func() (in module tvm.contrib.dlpack)
convert_to_expr() (in module tvm.relax)
ConvertBlocksToOpaque() (in module tvm.tir.transform)
ConvertForLoopsToSerial() (in module tvm.tir.transform)
ConvertLayout() (in module tvm.relax.transform)
ConvertSSA() (in module tvm.tir.transform)
ConvertToDataflow() (in module tvm.relax.transform)
convolution_inference() (in module tvm.contrib.nnpack)
convolution_inference_weight_transform() (in module tvm.contrib.nnpack)
convolution_inference_without_weight_transform() (in module tvm.contrib.nnpack)
copy() (tvm.tir.schedule.Schedule method)
copy_from_worker_0() (tvm.runtime.disco.Session method)
copy_to_worker_0() (tvm.runtime.disco.Session method)
copyfrom() (tvm.runtime.ndarray.NDArray method)
copysign() (in module tvm.tir)
copyto() (tvm.runtime.ndarray.NDArray method)
correlation_nchw() (in module tvm.topi.nn)
cos() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
cosh() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
CostModel (class in tvm.meta_schedule)
Count (class in tvm.runtime.profiling)
cpu() (in module tvm.runtime.ndarray)
(tvm.rpc.RPCSession method)
create() (in module tvm.target)
(tvm.meta_schedule.Builder static method)
(tvm.meta_schedule.CostModel static method)
(tvm.meta_schedule.Database static method)
(tvm.meta_schedule.FeatureExtractor static method)
(tvm.meta_schedule.MeasureCallback static method)
(tvm.meta_schedule.Mutator static method)
(tvm.meta_schedule.Postproc static method)
(tvm.meta_schedule.Runner static method)
(tvm.meta_schedule.ScheduleRule static method)
(tvm.meta_schedule.SearchStrategy static method)
(tvm.meta_schedule.SpaceGenerator static method)
(tvm.meta_schedule.TaskScheduler static method)
create_barriers() (in module tvm.tir)
create_dylib() (in module tvm.contrib.xcode)
create_empty() (tvm.relax.Function static method)
create_executable() (in module tvm.contrib.cc)
create_llvm() (in module tvm.contrib.clang)
create_prim_func() (in module tvm.te)
create_shared() (in module tvm.contrib.cc)
(in module tvm.contrib.ndk)
create_staticlib() (in module tvm.contrib.cc)
(in module tvm.contrib.ndk)
create_tvmjs_wasm() (in module tvm.contrib.emcc)
crop_and_resize() (in module tvm.topi.image)
cross_compiler() (in module tvm.contrib.cc)
cross_entropy_with_logits() (in module tvm.relax.op.nn)
csv() (tvm.runtime.profiling.Report method)
cuda() (in module tvm.runtime.ndarray)
(in module tvm.target)
(tvm.rpc.RPCSession method)
cumprod() (in module tvm.relax.op)
cumsum() (in module tvm.relax.op)
current() (tvm.meta_schedule.Database static method)
(tvm.meta_schedule.Profiler static method)
(tvm.relax.block_builder.BlockBuilder static method)
(tvm.target.Target static method)
(tvm.transform.PassContext static method)
current_block_is_dataflow() (tvm.relax.block_builder.BlockBuilder method)
D
Database (class in tvm.meta_schedule)
dataflow() (tvm.relax.block_builder.BlockBuilder method)
DataflowBlock (class in tvm.relax)
dataflowblock_pass() (in module tvm.relax.transform)
DataflowBlockPass (class in tvm.relax.transform)
DataflowBlockRewrite (class in tvm.relax)
DataflowScope (class in tvm.relax.block_builder)
DataflowUseInplaceCalls() (in module tvm.relax.transform)
DataflowVar (class in tvm.relax)
DataProducer (class in tvm.tir)
DataTypeImm (class in tvm.relax)
DeadCodeElimination() (in module tvm.relax.transform)
debug_copy_from() (tvm.runtime.disco.DRef method)
debug_get_from_remote() (tvm.runtime.disco.DRef method)
decisions (tvm.tir.schedule.Trace attribute)
decl_buffer() (in module tvm.tir)
declaration_conv2d_transpose_impl() (in module tvm.topi.nn)
declaration_conv3d_transpose_impl() (in module tvm.topi.nn)
declare_function() (tvm.relax.ExecBuilder method)
DeclBuffer (class in tvm.tir)
decompose_padding() (tvm.tir.schedule.Schedule method)
decompose_reduction() (tvm.tir.schedule.Schedule method)
DecomposeOpsForInference() (in module tvm.relax.transform)
DecomposeOpsForTraining() (in module tvm.relax.transform)
DecorateDeviceScope() (in module tvm.tir.transform)
DefaultGPUSchedule() (in module tvm.tir.transform)
definable_tir_vars_in_struct_info() (in module tvm.relax.analysis)
defined_symbolic_vars() (in module tvm.relax.analysis)
deformable_conv2d_nchw() (in module tvm.topi.nn)
deformable_conv2d_nhwc() (in module tvm.topi.nn)
dense() (in module tvm.topi.nn)
dense_pack() (in module tvm.topi.nn)
Dependency (class in tvm.tir.schedule)
DepKind (class in tvm.tir.schedule)
depth_to_space() (in module tvm.topi.nn)
depthwise_conv2d_backward_input_nhwc() (in module tvm.topi.nn)
depthwise_conv2d_backward_weight_nhwc() (in module tvm.topi.nn)
depthwise_conv2d_nchw() (in module tvm.topi.nn)
depthwise_conv2d_NCHWc() (in module tvm.topi.nn)
depthwise_conv2d_nhwc() (in module tvm.topi.nn)
dequantize() (in module tvm.relax.op)
derive_call_ret_struct_info() (in module tvm.relax.analysis)
derived_object() (in module tvm.meta_schedule)
detach_params() (in module tvm.relax.frontend)
detect_buffer_access_lca() (in module tvm.tir.analysis.analysis)
detect_recursion() (in module tvm.relax.analysis)
device (tvm.runtime.ndarray.NDArray property)
device() (in module tvm.runtime.ndarray)
(tvm.rpc.RPCSession method)
device_metrics (tvm.runtime.profiling.Report attribute)
DeviceWrapper (class in tvm.runtime.profiling)
DiagnosticError
DictAttrs (class in tvm.ir)
dilate() (in module tvm.topi.nn)
dilation2d_nchw() (in module tvm.topi.image)
dilation2d_nhwc() (in module tvm.topi.image)
dilation_h (tvm.topi.nn.Workload attribute)
dilation_w (tvm.topi.nn.Workload attribute)
DirectoryCreatedPastAtExit
Div (class in tvm.tir)
div() (in module tvm.te)
(in module tvm.tir)
divide() (in module tvm.relax.op)
DModule (class in tvm.runtime.disco)
dom (tvm.dlight.IterInfo property)
dom() (tvm.dlight.BlockInfo method)
dom_kind() (tvm.dlight.BlockInfo method)
download() (tvm.rpc.RPCSession method)
download_linked_module() (tvm.rpc.RPCSession method)
dp4a() (in module tvm.tir)
DPackedFunc (class in tvm.runtime.disco)
DRef (class in tvm.runtime.disco)
dropout() (in module tvm.relax.op.nn)
DropoutAttrs (class in tvm.relax.op.op_attrs)
dtype (tvm.runtime.ndarray.NDArray property)
(tvm.te.TensorSlice property)
DummyGlobalInfo (class in tvm.ir)
dump_pruned() (tvm.meta_schedule.Database method)
Duration (class in tvm.runtime.profiling)
dynamic_strided_slice() (in module tvm.relax.op)
dynamo_capture_subgraphs() (in module tvm.relax.frontend.torch)
E
einsum() (in module tvm.relax.op)
EinsumAttrs (class in tvm.relax.op.op_attrs)
EliminateCommonSubexpr() (in module tvm.relax.transform)
emit() (tvm.relax.block_builder.BlockBuilder method)
emit_call() (tvm.relax.ExecBuilder method)
emit_func_output() (tvm.relax.block_builder.BlockBuilder method)
emit_goto() (tvm.relax.ExecBuilder method)
emit_if() (tvm.relax.ExecBuilder method)
emit_normalized() (tvm.relax.block_builder.BlockBuilder method)
emit_output() (tvm.relax.block_builder.BlockBuilder method)
emit_ret() (tvm.relax.ExecBuilder method)
emit_te() (tvm.relax.block_builder.BlockBuilder method)
empty() (in module tvm.runtime.ndarray)
(tvm.runtime.disco.Session method)
end_checkpoint() (in module tvm.relax.op.grad)
end_profile_intrinsic() (in module tvm.tir)
end_scope() (tvm.relax.block_builder.BlockBuilder method)
ensure_zero_offset() (in module tvm.relax.op.memory)
enter_pass_ctx() (tvm.instrument.PassInstrument method)
enter_postproc() (tvm.tir.schedule.Schedule method)
EnvFunc (class in tvm.ir)
EQ (class in tvm.tir)
equal() (in module tvm.relax.op)
equal_const_int() (in module tvm.topi.nn)
erase_to_well_defined() (in module tvm.relax.analysis)
erf() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
estimate_memory_usage() (in module tvm.relax.analysis)
estimate_tir_flops() (in module tvm.tir.analysis.analysis)
Evaluate (class in tvm.tir)
ewise_fma() (in module tvm.relax.op)
ExecBuilder (class in tvm.relax)
exit_pass_ctx() (tvm.instrument.PassInstrument method)
exp() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
exp10() (in module tvm.tir)
exp2() (in module tvm.tir)
expand_dims() (in module tvm.relax.op)
ExpandDimsAttrs (class in tvm.relax.op.op_attrs)
ExpandMatmulOfSum() (in module tvm.relax.transform)
ExpandTupleArguments() (in module tvm.relax.transform)
Expr (in module tvm.relax)
(in module tvm.relax.op.grad)
expr_deep_equal() (in module tvm.tir.analysis.analysis)
ExprFunctor (class in tvm.relax)
ExprRV (in module tvm.tir.schedule)
ext_dev() (in module tvm.runtime.ndarray)
(tvm.rpc.RPCSession method)
extern() (in module tvm.relax)
(in module tvm.te)
extern_primfunc() (in module tvm.te)
ExternFunc (class in tvm.relax)
ExternOp (class in tvm.te)
extract_from() (tvm.meta_schedule.FeatureExtractor method)
ExtractedTask (class in tvm.meta_schedule)
ExtractPrimFuncConstants() (in module tvm.tir.transform)
eye() (in module tvm.relax.op)
eye_like() (in module tvm.relax.op)
F
f() (tvm.relax.ExecBuilder method)
factor_of() (tvm.tir.Layout method)
fast_softmax() (in module tvm.topi.nn)
FastMathTransform (class in tvm.relax.transform)
FeatureExtractor (class in tvm.meta_schedule)
FewShotTuning() (in module tvm.relax.transform)
fifo_buffer() (in module tvm.topi.nn)
FileLock (class in tvm.contrib.utils)
filelock() (in module tvm.contrib.utils)
Filter() (in module tvm.tir.transform)
finalize() (tvm.relax.block_builder.BlockBuilder method)
find_anchor_block() (in module tvm.tir.analysis.analysis)
find_clang() (in module tvm.contrib.clang)
find_cuda_path() (in module tvm.contrib.nvcc)
find_lld() (in module tvm.contrib.rocm)
find_rocm_path() (in module tvm.contrib.rocm)
flatten() (in module tvm.relax.op)
(in module tvm.topi.nn)
FlattenBuffer() (in module tvm.tir.transform)
flip() (in module tvm.relax.op)
FlipAttrs (class in tvm.relax.op.op_attrs)
FloatImm (class in tvm.tir)
floor() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
floor_divide() (in module tvm.relax.op)
floor_mod() (in module tvm.relax.op)
FloorDiv (class in tvm.tir)
floordiv() (in module tvm.te)
(in module tvm.tir)
FloorMod (class in tvm.tir)
floormod() (in module tvm.te)
(in module tvm.tir)
fmod() (in module tvm.te)
(in module tvm.tir)
FoldBatchnormToConv2D (class in tvm.relax.transform)
FoldConstant() (in module tvm.relax.transform)
For (class in tvm.tir)
ForceNarrowIndexToInt32() (in module tvm.tir.transform)
fork_seed() (tvm.tir.schedule.Schedule method)
ForKind (class in tvm.tir)
forward_index() (tvm.tir.BijectiveLayout method)
forward_shape() (tvm.tir.BijectiveLayout method)
FP8ComputeLegalize() (in module tvm.tir.transform)
FP8StorageLegalize() (in module tvm.tir.transform)
free_symbolic_vars() (in module tvm.relax.analysis)
free_vars() (in module tvm.relax.analysis)
from_callable() (tvm.dlight.ScheduleRule static method)
from_device() (tvm.target.Target static method)
from_dlpack() (in module tvm.runtime.ndarray)
from_exported_program() (in module tvm.relax.frontend.torch)
from_expr() (tvm.ir.IRModule static method)
from_func() (tvm.tir.IndexMap static method)
from_func_with_separators() (tvm.tir.IndexMap static method)
from_fx() (in module tvm.relax.frontend.torch)
from_json() (tvm.runtime.profiling.Report class method)
from_min_extent() (tvm.ir.Range static method)
from_onnx() (in module tvm.relax.frontend.onnx)
from_stablehlo() (in module tvm.relax.frontend.stablehlo)
full() (in module tvm.relax.op)
full_like() (in module tvm.relax.op)
fully_connected_inference() (in module tvm.contrib.nnpack)
func_working_on (tvm.tir.schedule.Schedule property)
FuncStructInfo (class in tvm.relax)
Function (class in tvm.relax)
function() (tvm.relax.block_builder.BlockBuilder method)
(tvm.relax.ExecBuilder method)
function_pass() (in module tvm.relax.transform)
FunctionPass (class in tvm.relax.transform)
functions_items() (tvm.ir.IRModule method)
FunctionScope (class in tvm.relax.block_builder)
FuncType (class in tvm.ir)
fuse() (tvm.tir.schedule.Schedule method)
FuseOps() (in module tvm.relax.transform)
FuseOpsByPattern() (in module tvm.relax.transform)
FuseTIR() (in module tvm.relax.transform)
FuseTransposeMatmul (class in tvm.relax.transform)
FusionPattern (class in tvm.relax.transform)
G
gather_elements() (in module tvm.relax.op)
gather_nd() (in module tvm.relax.op)
gather_to_worker0() (tvm.runtime.disco.Session method)
GE (class in tvm.tir)
gelu() (in module tvm.relax.op.nn)
gelu_tanh() (in module tvm.relax.op.nn)
generate_design_space() (tvm.meta_schedule.SpaceGenerator method)
(tvm.meta_schedule.TuneContext method)
generate_measure_candidates() (tvm.meta_schedule.SearchStrategy method)
(tvm.meta_schedule.TuneContext method)
get() (tvm.ir.DictAttrs method)
(tvm.ir.EnvFunc static method)
(tvm.ir.Map method)
(tvm.ir.Op static method)
(tvm.meta_schedule.Profiler method)
(tvm.relax.block_builder.BlockBuilder method)
(tvm.relax.ExecBuilder method)
(tvm.tir.schedule.InstructionKind static method)
(tvm.tir.schedule.Schedule method)
(tvm.tir.TensorIntrin static method)
get_1d_indices() (in module tvm.topi.image)
get_1d_pixel() (in module tvm.topi.image)
get_2d_indices() (in module tvm.topi.image)
get_2d_pixel() (in module tvm.topi.image)
get_3d_indices() (in module tvm.topi.image)
get_3d_pixel() (in module tvm.topi.image)
get_active_lane_mask() (in module tvm.tir)
get_all_tuning_records() (tvm.meta_schedule.Database method)
get_attr() (tvm.ir.IRModule method)
(tvm.ir.Op method)
get_block() (tvm.tir.schedule.Schedule method)
get_block_access_region() (in module tvm.tir.analysis.analysis)
get_block_read_write_region() (in module tvm.tir.analysis.analysis)
get_block_scope() (tvm.tir.BlockDependenceInfo method)
(tvm.tir.schedule.ScheduleState method)
get_cc() (in module tvm.contrib.cc)
get_child_blocks() (tvm.tir.schedule.Schedule method)
get_closest_index() (in module tvm.topi.image)
get_const_int() (in module tvm.topi.nn)
get_const_tuple() (in module tvm.topi.nn)
get_consumers() (tvm.tir.schedule.Schedule method)
get_cuda_version() (in module tvm.contrib.nvcc)
get_current_trace() (tvm.transform.PassContext method)
get_decision() (tvm.tir.schedule.Trace method)
get_default_pipeline() (in module tvm.relax)
get_default_tir_pipeline() (in module tvm.tir)
get_deps_by_dst() (tvm.tir.schedule.BlockScope method)
get_deps_by_src() (tvm.tir.schedule.BlockScope method)
get_flattened_buffer() (tvm.tir.Buffer method)
get_function() (tvm.rpc.RPCSession method)
get_global_func() (tvm.runtime.disco.Session method)
get_global_symbol_section_map() (in module tvm.contrib.cc)
(in module tvm.contrib.ndk)
get_global_var() (tvm.ir.IRModule method)
get_global_vars() (tvm.ir.IRModule method)
get_int() (tvm.ir.Attrs method)
get_int_tuple() (tvm.ir.Attrs method)
get_inx() (in module tvm.topi.image)
get_kind_attr() (tvm.target.Target method)
get_loops() (tvm.tir.schedule.Schedule method)
get_output_blocks() (tvm.tir.schedule.Schedule method)
get_outputs() (tvm.relax.VirtualMachine method)
(tvm.runtime.relax_vm.VirtualMachine method)
get_pad_tuple() (in module tvm.topi.image)
(in module tvm.topi.nn)
get_pad_tuple1d() (in module tvm.topi.nn)
get_pad_tuple3d() (in module tvm.topi.nn)
get_pad_tuple_generic() (in module tvm.topi.nn)
get_pipeline() (in module tvm.relax)
get_producers() (tvm.tir.schedule.Schedule method)
get_shape_of() (in module tvm.relax)
get_sref() (tvm.tir.BlockDependenceInfo method)
(tvm.tir.schedule.Schedule method)
(tvm.tir.schedule.ScheduleState method)
get_static_type() (in module tvm.relax.analysis)
get_str() (tvm.ir.Attrs method)
get_target_by_dump_machine() (in module tvm.contrib.cc)
get_target_device_type() (tvm.target.Target method)
get_tir_pipeline() (in module tvm.tir)
get_top_k() (tvm.meta_schedule.Database method)
get_trace_stack() (tvm.transform.PassContext method)
get_trace_stack_size() (tvm.transform.PassContext method)
get_tuning_api_database() (tvm.transform.PassContext method)
get_unique_name() (tvm.relax.block_builder.BlockBuilder method)
get_var2val() (in module tvm.relax.analysis)
get_var_remap() (tvm.relax.PyExprMutator method)
get_vscale_expr() (in module tvm.tir)
get_vtcm_compaction_passes() (in module tvm.tir.analysis.analysis)
global_pool() (in module tvm.topi.nn)
GlobalInfo (class in tvm.ir)
GlobalVar (class in tvm.ir)
gpu() (in module tvm.runtime.ndarray)
Gradient() (in module tvm.relax.transform)
greater() (in module tvm.relax.op)
greater_equal() (in module tvm.relax.op)
grid_sample() (in module tvm.topi.image)
group_conv1d_ncw() (in module tvm.topi.nn)
group_conv1d_nwc() (in module tvm.topi.nn)
group_conv1d_transpose_ncw() (in module tvm.topi.nn)
group_conv2d_nchw() (in module tvm.topi.nn)
group_conv2d_nhwc() (in module tvm.topi.nn)
group_conv2d_transpose_nchw() (in module tvm.topi.nn)
group_conv3d_transpose_ncdhw() (in module tvm.topi.nn)
group_norm() (in module tvm.relax.op.nn)
(in module tvm.topi.nn)
GT (class in tvm.tir)
H
has_attr() (tvm.ir.Op method)
has_reshape_pattern() (in module tvm.relax.analysis)
has_workload() (tvm.meta_schedule.Database method)
have_cudagraph() (in module tvm.contrib.nvcc)
have_fp16() (in module tvm.contrib.nvcc)
have_int8() (in module tvm.contrib.nvcc)
have_matrixcore() (in module tvm.contrib.rocm)
have_tensorcore() (in module tvm.contrib.nvcc)
height (tvm.topi.nn.Workload attribute)
hexagon() (in module tvm.runtime.ndarray)
(in module tvm.target)
(tvm.rpc.RPCSession method)
hint_on_device() (in module tvm.relax.op)
HoistedConditionals (class in tvm.tir.transform)
HoistedLetBindings (class in tvm.tir.transform)
HoistExpression() (in module tvm.tir.transform)
HoistIfThenElse() (in module tvm.tir.transform)
hypot() (in module tvm.tir)
I
Id (class in tvm.relax)
If (class in tvm.relax)
if_then_else() (in module tvm.te)
(in module tvm.tir)
IfElseExpr (tvm.tir.transform.HoistedConditionals attribute)
IfElseStmt (tvm.tir.transform.HoistedConditionals attribute)
IfThenElse (class in tvm.tir)
ignore_loop_partition() (in module tvm.tir)
imm() (tvm.relax.ExecBuilder method)
import_python_module() (tvm.runtime.disco.Session method)
in_dtype (tvm.topi.nn.Workload attribute)
in_filter (tvm.topi.nn.Workload attribute)
inc_num_evals() (tvm.transform.PassContext method)
index_of() (tvm.tir.Layout method)
indexdiv() (in module tvm.te)
(in module tvm.tir)
IndexMap (class in tvm.tir)
indexmod() (in module tvm.te)
(in module tvm.tir)
InferFragment() (in module tvm.tir.transform)
infinity() (in module tvm.tir)
info (tvm.transform.Pass property)
init_ccl() (tvm.runtime.disco.Session method)
InitAttrs (class in tvm.relax.op.op_attrs)
InjectDoubleBuffer() (in module tvm.tir.transform)
InjectPermutedLayout() (in module tvm.tir.transform)
InjectPTXAsyncCopy() (in module tvm.tir.transform)
InjectPTXLDG32() (in module tvm.tir.transform)
InjectRollingBuffer() (in module tvm.tir.transform)
InjectSoftwarePipeline() (in module tvm.tir.transform)
InjectVirtualThread() (in module tvm.tir.transform)
inline_mark() (tvm.tir.schedule.StmtSRef static method)
InlinePrivateFunctions() (in module tvm.relax.transform)
(in module tvm.tir.transform)
inputs (tvm.tir.schedule.Instruction attribute)
instance_norm() (in module tvm.topi.nn)
Instruction (class in tvm.tir.schedule)
InstructionKind (class in tvm.tir.schedule)
InstrumentBoundCheckers() (in module tvm.tir.transform)
InstrumentProfileIntrinsics() (in module tvm.tir.transform)
insts (tvm.tir.schedule.Trace attribute)
intel_graphics() (in module tvm.target)
InternalError
IntImm (class in tvm.tir)
inverse() (tvm.tir.IndexMap method)
invoke_closure() (in module tvm.relax.op)
(tvm.relax.VirtualMachine method)
(tvm.runtime.relax_vm.VirtualMachine method)
invoke_pure_closure() (in module tvm.relax.op)
invoke_stateful() (tvm.relax.VirtualMachine method)
(tvm.runtime.relax_vm.VirtualMachine method)
IPCAllReduceRewrite (class in tvm.relax.transform)
ir_transform() (in module tvm.tir.stmt_functor)
IRModule (class in tvm.ir)
is_available() (in module tvm.contrib.nnpack)
is_base_of() (tvm.relax.StructInfo method)
is_elementwise() (tvm.dlight.BlockInfo method)
is_equivalent_to() (tvm.tir.IndexMap method)
is_gemm() (tvm.dlight.BlockInfo method)
is_gemv() (tvm.dlight.BlockInfo method)
is_injective() (tvm.dlight.BlockInfo method)
is_pure (tvm.tir.schedule.InstructionKind property)
is_pure_function() (in module tvm.tir.analysis.analysis)
is_reduction() (tvm.dlight.BlockInfo method)
is_source_path() (in module tvm.contrib.utils)
is_target_available() (tvm.dlight.ScheduleRule method)
isfinite() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
isinf() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
isnan() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
isnullptr() (in module tvm.tir)
items() (tvm.ir.DictAttrs method)
(tvm.ir.Map method)
IterInfo (class in tvm.dlight)
IterVar (class in tvm.tir)
J
join_running_task() (tvm.meta_schedule.TaskScheduler method)
json() (tvm.runtime.profiling.Report method)
K
kernel_h (tvm.topi.nn.Workload attribute)
kernel_w (tvm.topi.nn.Workload attribute)
keys() (tvm.ir.Attrs method)
(tvm.ir.DictAttrs method)
kill_storage() (in module tvm.relax.op.memory)
kill_tensor() (in module tvm.relax.op.memory)
KillAfterLastUse() (in module tvm.relax.transform)
kind (tvm.tir.schedule.Instruction attribute)
L
LambdaLift() (in module tvm.relax.transform)
layer_norm() (in module tvm.relax.op.nn)
(in module tvm.topi.nn)
LayerNormAttrs (class in tvm.relax.op.op_attrs)
Layout (class in tvm.tir)
layout() (in module tvm.tir)
layout_transform() (in module tvm.relax.op)
LayoutTransformAttrs (class in tvm.relax.op.op_attrs)
LazyGetInput() (in module tvm.relax.transform)
LazySetOutput() (in module tvm.relax.transform)
LazyTransformParams (class in tvm.relax.transform)
ldexp() (in module tvm.tir)
LE (class in tvm.tir)
leaky_relu() (in module tvm.topi.nn)
leakyrelu() (in module tvm.relax.op.nn)
left_shift() (in module tvm.relax.op)
LegalizeOps() (in module tvm.relax.transform)
LegalizePackedCalls() (in module tvm.tir.transform)
less() (in module tvm.relax.op)
less_equal() (in module tvm.relax.op)
Let (class in tvm.tir)
LetExpr (tvm.tir.transform.HoistedLetBindings attribute)
LetStmt (class in tvm.tir)
(tvm.tir.transform.HoistedLetBindings attribute)
LiftThreadBinding() (in module tvm.tir.transform)
LiftTransformParams() (in module tvm.relax.transform)
likely() (in module tvm.tir)
linear() (in module tvm.relax.op)
list_configs() (tvm.transform.PassContext static method)
list_field_info() (tvm.ir.Attrs method)
list_kinds() (tvm.target.Target static method)
list_op_names() (tvm.ir.Op static method)
list_tags() (in module tvm.target)
listdir() (tvm.contrib.utils.TempDirectory method)
(tvm.rpc.RPCSession method)
load() (tvm.meta_schedule.CostModel method)
load_json() (in module tvm.ir)
load_module() (tvm.rpc.RPCSession method)
load_vm_module() (tvm.runtime.disco.Session method)
LocalSession (class in tvm.rpc)
log() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
log10() (in module tvm.te)
(in module tvm.tir)
log1p() (in module tvm.tir)
log2() (in module tvm.te)
(in module tvm.tir)
log_softmax() (in module tvm.contrib.miopen)
(in module tvm.relax.op.nn)
(in module tvm.topi.nn)
logical_and() (in module tvm.relax.op)
logical_not() (in module tvm.relax.op)
logical_or() (in module tvm.relax.op)
logical_xor() (in module tvm.relax.op)
lookup_binding() (tvm.relax.block_builder.BlockBuilder method)
(tvm.relax.PyExprMutator method)
lookup_param() (in module tvm.tir)
loop_partition() (tvm.tir.schedule.Schedule method)
LoopPartition() (in module tvm.tir.transform)
LoopRV (class in tvm.tir.schedule)
LowerAllocTensor() (in module tvm.relax.transform)
LowerAsyncDMA() (in module tvm.tir.transform)
LowerAutoCopy() (in module tvm.tir.transform)
LowerCrossThreadReduction() (in module tvm.tir.transform)
LowerCustomDatatypes() (in module tvm.tir.transform)
LowerDeviceKernelLaunch() (in module tvm.tir.transform)
LowerDeviceStorageAccessInfo() (in module tvm.tir.transform)
LowerGPUIPCAllocStorage (class in tvm.relax.transform)
LowerInitBlock() (in module tvm.tir.transform)
LowerIntrin() (in module tvm.tir.transform)
LowerMatchBuffer() (in module tvm.tir.transform)
LowerOpaqueBlock() (in module tvm.tir.transform)
LowerRuntimeBuiltin() (in module tvm.relax.transform)
LowerThreadAllreduce() (in module tvm.tir.transform)
LowerTVMBuiltin() (in module tvm.tir.transform)
LowerVtcmAlloc() (in module tvm.tir.transform)
LowerWarpMemory() (in module tvm.tir.transform)
lrn() (in module tvm.topi.nn)
lstm() (in module tvm.topi.nn)
LT (class in tvm.tir)
M
make_closure() (in module tvm.relax.op)
make_compilation_config() (in module tvm.target)
make_filled_simdgroup_matrix() (in module tvm.tir)
make_node() (in module tvm.ir)
MakePackedAPI() (in module tvm.tir.transform)
MakeUnpackedAPI() (in module tvm.tir.transform)
mali() (in module tvm.target)
ManifestSharedMemoryLocalStage() (in module tvm.tir.transform)
Map (class in tvm.ir)
map_indices() (tvm.tir.IndexMap method)
map_ndarray() (tvm.tir.IndexMap method)
map_shape() (tvm.tir.IndexMap method)
masked_fill() (in module tvm.relax.op)
match_cast() (tvm.relax.block_builder.BlockBuilder method)
MatchBufferRegion (class in tvm.tir)
MatchCast (class in tvm.relax)
matmul() (in module tvm.contrib.cblas)
(in module tvm.contrib.cublas)
(in module tvm.contrib.rocblas)
(in module tvm.relax.op)
(in module tvm.topi.nn)
MatmulAttrs (class in tvm.relax.op.op_attrs)
mattr (tvm.target.Target property)
Max (class in tvm.tir)
max() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
max_block_size_x (tvm.target.Target property)
max_block_size_y (tvm.target.Target property)
max_num_threads (tvm.target.Target property)
max_pool1d() (in module tvm.relax.op.nn)
max_pool2d() (in module tvm.relax.op.nn)
max_pool2d_backward() (in module tvm.relax.op.grad)
max_pool3d() (in module tvm.relax.op.nn)
max_value() (in module tvm.te)
(in module tvm.tir)
maximum() (in module tvm.relax.op)
mcpu (tvm.target.Target property)
mean() (in module tvm.relax.op)
MeasureCallback (class in tvm.meta_schedule)
MeasureCandidate (class in tvm.meta_schedule)
memoize() (in module tvm.contrib.pickle_memoize)
merge() (tvm.tir.schedule.Schedule method)
MergeCompositeFunctions() (in module tvm.relax.transform)
MergeSharedMemoryAllocations() (in module tvm.tir.transform)
metal() (in module tvm.runtime.ndarray)
(tvm.rpc.RPCSession method)
MetaScheduleApplyDatabase() (in module tvm.relax.transform)
MetaScheduleTuneIRMod() (in module tvm.relax.transform)
MetaScheduleTuneTIR() (in module tvm.relax.transform)
MetricCollector (class in tvm.runtime.profiling)
Min (class in tvm.tir)
min() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
min_value() (in module tvm.te)
(in module tvm.tir)
minimum() (in module tvm.relax.op)
mirror_pad() (in module tvm.topi.nn)
mma_fill() (in module tvm.tir)
mma_store() (in module tvm.tir)
Mod (class in tvm.tir)
mod (tvm.tir.schedule.Schedule property)
mod() (in module tvm.relax.op)
model (tvm.target.Target property)
module
tvm.contrib
tvm.contrib.cblas
tvm.contrib.cc
tvm.contrib.clang
tvm.contrib.cublas
tvm.contrib.dlpack
tvm.contrib.emcc
tvm.contrib.miopen
tvm.contrib.ndk
tvm.contrib.nnpack
tvm.contrib.nvcc
tvm.contrib.pickle_memoize
tvm.contrib.random
tvm.contrib.rocblas
tvm.contrib.rocm
tvm.contrib.spirv
tvm.contrib.tar
tvm.contrib.utils
tvm.contrib.xcode
tvm.dlight
tvm.driver
tvm.error
tvm.instrument
tvm.ir
tvm.meta_schedule
tvm.relax
tvm.relax.analysis
tvm.relax.block_builder
tvm.relax.frontend
tvm.relax.frontend.onnx
tvm.relax.frontend.stablehlo
tvm.relax.frontend.torch
tvm.relax.op
tvm.relax.op.builtin
tvm.relax.op.ccl
tvm.relax.op.distributed
tvm.relax.op.grad
tvm.relax.op.image
tvm.relax.op.memory
tvm.relax.op.nn
tvm.relax.op.op_attrs
tvm.relax.transform
tvm.rpc
tvm.runtime
tvm.runtime.disco
tvm.runtime.ndarray
tvm.runtime.profiling
tvm.runtime.relax_vm
tvm.target
tvm.te
tvm.tir
tvm.tir.analysis.analysis
tvm.tir.schedule
tvm.tir.stmt_functor
tvm.tir.transform
tvm.topi.image
tvm.topi.nn
tvm.transform
module_pass() (in module tvm.transform)
ModulePass (class in tvm.transform)
mtl() (in module tvm.runtime.ndarray)
Mul (class in tvm.tir)
multinomial_from_uniform() (in module tvm.relax.op)
multiply() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
mutate_irmodule() (tvm.relax.DataflowBlockRewrite method)
mutated_dfb() (tvm.relax.DataflowBlockRewrite method)
mutated_root_fn() (tvm.relax.DataflowBlockRewrite method)
Mutator (class in tvm.meta_schedule)
N
name (tvm.tir.schedule.InstructionKind attribute)
name_hint (tvm.relax.Var property)
name_to_binding() (in module tvm.relax.analysis)
namedtuple() (in module tvm.topi.nn)
NarrowDataType() (in module tvm.tir.transform)
nchw_pack_layout() (in module tvm.topi.image)
nchw_xc_layout() (in module tvm.topi.image)
NDArray (class in tvm.runtime.ndarray)
ndim (tvm.te.Tensor property)
NE (class in tvm.tir)
nearbyint() (in module tvm.te)
(in module tvm.tir)
negative() (in module tvm.relax.op)
Never (tvm.tir.transform.HoistedConditionals attribute)
(tvm.tir.transform.HoistedLetBindings attribute)
next_task_id() (tvm.meta_schedule.TaskScheduler method)
nextafter() (in module tvm.tir)
nll_loss() (in module tvm.relax.op.nn)
(in module tvm.topi.nn)
nll_loss_backward() (in module tvm.relax.op.grad)
no_grad() (in module tvm.relax.op.grad)
Node (class in tvm.ir)
non_surjective_inverse() (tvm.tir.IndexMap method)
nonzero() (in module tvm.relax.op)
normal() (in module tvm.contrib.random)
Normalize() (in module tvm.relax.transform)
normalize() (tvm.relax.block_builder.BlockBuilder method)
normalize_file_list_by_unpacking_tars() (in module tvm.contrib.tar)
normalize_prim_func() (in module tvm.dlight)
NormalizeGlobalVar() (in module tvm.relax.transform)
Not (class in tvm.tir)
not_equal() (in module tvm.relax.op)
notify_runner_results() (tvm.meta_schedule.SearchStrategy method)
(tvm.meta_schedule.TuneContext method)
null_value() (in module tvm.relax.op)
num_workers (tvm.runtime.disco.Session property)
numpy() (tvm.runtime.ndarray.NDArray method)
numpyasarray() (in module tvm.runtime.ndarray)
O
ObjectStructInfo (class in tvm.relax)
ObjectType (class in tvm.relax)
offset_of() (tvm.tir.Buffer method)
one_hot() (in module tvm.relax.op)
ones() (in module tvm.relax.op)
ones_like() (in module tvm.relax.op)
OOBChecker() (in module tvm.tir.analysis.analysis)
Op (class in tvm.ir)
op (tvm.te.Tensor property)
OPAQUE (tvm.tir.schedule.DepKind attribute)
opaque_func() (tvm.relax.FuncStructInfo static method)
OpAttributeInvalid
OpAttributeRequired
OpAttributeUnImplemented
opencl() (in module tvm.runtime.ndarray)
OpError
OpNotImplemented
optimize() (in module tvm.contrib.spirv)
OptimizeLayoutTransform (class in tvm.relax.transform)
options (tvm.target.TargetKind property)
options_from_name() (tvm.target.TargetKind static method)
Or (class in tvm.tir)
out_dtype (tvm.topi.nn.Workload attribute)
out_filter (tvm.topi.nn.Workload attribute)
outputs (tvm.tir.schedule.Instruction attribute)
override_instruments() (tvm.transform.PassContext method)
P
PackedFuncType (class in tvm.relax)
pad() (in module tvm.relax.op.nn)
(in module tvm.topi.image)
(in module tvm.topi.nn)
pad_einsum() (tvm.tir.schedule.Schedule method)
padb (tvm.topi.nn.Workload attribute)
padl (tvm.topi.nn.Workload attribute)
padr (tvm.topi.nn.Workload attribute)
padt (tvm.topi.nn.Workload attribute)
parallel() (tvm.tir.schedule.Schedule method)
parent (tvm.tir.schedule.StmtSRef property)
parse_compute_version() (in module tvm.contrib.nvcc)
(in module tvm.contrib.rocm)
Pass (class in tvm.transform)
pass_instrument() (in module tvm.instrument)
PassContext (class in tvm.transform)
PassInfo (class in tvm.transform)
PassInstrument (class in tvm.instrument)
PassPrintingInstrument (class in tvm.instrument)
PassTimingInstrument (class in tvm.instrument)
PatternCheckContext (class in tvm.relax.transform)
Percent (class in tvm.runtime.profiling)
permute_dims() (in module tvm.relax.op)
PermuteDimsAttrs (class in tvm.relax.op.op_attrs)
placeholder() (in module tvm.te)
PlaceholderOp (class in tvm.te)
PlanAndUpdateBufferAllocationLocation() (in module tvm.tir.transform)
PointerType (class in tvm.ir)
PointerValueTypeRewrite() (in module tvm.tir.transform)
pool1d() (in module tvm.topi.nn)
pool2d() (in module tvm.topi.nn)
Pool2DAttrs (class in tvm.relax.op.op_attrs)
pool3d() (in module tvm.topi.nn)
pool_grad() (in module tvm.topi.nn)
pop() (tvm.tir.schedule.Trace method)
pop_trace() (tvm.transform.PassContext method)
popcount() (in module tvm.te)
(in module tvm.tir)
PopenSession (class in tvm.rpc)
post_order_visit() (in module tvm.relax.analysis)
(in module tvm.tir.stmt_functor)
post_tuning() (tvm.meta_schedule.SearchStrategy method)
(tvm.meta_schedule.TuneContext method)
Postproc (class in tvm.meta_schedule)
pow() (in module tvm.tir)
power() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
pre_order_visit() (in module tvm.tir.stmt_functor)
pre_tuning() (tvm.meta_schedule.SearchStrategy method)
(tvm.meta_schedule.TuneContext method)
predict() (tvm.meta_schedule.CostModel method)
Prefetch (class in tvm.tir)
prelu() (in module tvm.topi.nn)
prim_func_pass() (in module tvm.tir.transform)
PrimExpr (class in tvm.ir)
PrimFunc (class in tvm.tir)
PrimFuncPass (class in tvm.tir.transform)
PrimStructInfo (class in tvm.relax)
PrimType (class in tvm.ir)
PrimValue (class in tvm.relax)
print() (in module tvm.relax.op)
print_tuning_statistics() (tvm.meta_schedule.TaskScheduler method)
PrintAfterAll (class in tvm.instrument)
PrintBeforeAll (class in tvm.instrument)
PrintIR() (in module tvm.transform)
ProcessSession (class in tvm.runtime.disco)
prod() (in module tvm.relax.op)
ProducerLoad (class in tvm.tir)
ProducerRealize (class in tvm.tir)
ProducerStore (class in tvm.tir)
profile() (tvm.relax.VirtualMachine method)
(tvm.runtime.relax_vm.VirtualMachine method)
profile_function() (in module tvm.runtime.profiling)
Profiler (class in tvm.meta_schedule)
ptx_arrive_barrier() (in module tvm.tir)
ptx_arrive_barrier_expect_tx() (in module tvm.tir)
ptx_commit_group() (in module tvm.tir)
ptx_cp_async() (in module tvm.tir)
ptx_cp_async_barrier() (in module tvm.tir)
ptx_cp_async_bulk() (in module tvm.tir)
ptx_init_barrier_thread_count() (in module tvm.tir)
ptx_ldmatrix() (in module tvm.tir)
ptx_mma() (in module tvm.tir)
ptx_mma_sp() (in module tvm.tir)
ptx_wait_barrier() (in module tvm.tir)
ptx_wait_group() (in module tvm.tir)
push_trace() (tvm.transform.PassContext method)
PyExprMutator (class in tvm.relax)
PyExprVisitor (class in tvm.relax)
Q
q_multiply_shift() (in module tvm.tir)
q_multiply_shift_per_axis() (in module tvm.tir)
quantize() (in module tvm.relax.op)
query() (tvm.meta_schedule.Database method)
query_ir_module() (tvm.meta_schedule.Database method)
query_schedule() (tvm.meta_schedule.Database method)
query_tuning_record() (tvm.meta_schedule.Database method)
R
r() (tvm.relax.ExecBuilder method)
Ramp (class in tvm.tir)
randint() (in module tvm.contrib.random)
Range (class in tvm.ir)
rasp() (in module tvm.target)
Ratio (class in tvm.runtime.profiling)
RAW (tvm.tir.schedule.DepKind attribute)
RealizeVDevice() (in module tvm.relax.transform)
redistribute() (in module tvm.relax.op.distributed)
redistribute_replica_to_shard() (in module tvm.relax.op.distributed)
Reduce (class in tvm.tir)
reduce() (in module tvm.topi.nn)
reduce_axis() (in module tvm.te)
ReduceBranchingThroughOvercompute() (in module tvm.tir.transform)
register() (tvm.tir.TensorIntrin static method)
register_error() (in module tvm.error)
register_gradient() (in module tvm.relax.op)
register_intrin_lowering() (in module tvm.ir)
register_op_attr() (in module tvm.ir)
register_pipeline() (in module tvm.relax)
reindex() (tvm.tir.schedule.Schedule method)
reindex_cache_read() (tvm.tir.schedule.Schedule method)
reindex_cache_write() (tvm.tir.schedule.Schedule method)
reinterpret() (in module tvm.tir)
relax_dynamo() (in module tvm.relax.frontend.torch)
RelaxExpr (class in tvm.ir)
release() (tvm.contrib.utils.FileLock method)
relpath() (tvm.contrib.utils.TempDirectory method)
relu() (in module tvm.relax.op.nn)
(in module tvm.topi.nn)
remove() (tvm.contrib.utils.TempDirectory method)
(tvm.rpc.RPCSession method)
remove_all_unused() (in module tvm.relax.analysis)
(tvm.relax.DataflowBlockRewrite method)
remove_rv() (tvm.tir.schedule.Schedule method)
remove_unused() (tvm.relax.DataflowBlockRewrite method)
RemoveAssume() (in module tvm.tir.transform)
RemoveNoOp() (in module tvm.tir.transform)
RemovePurityChecking() (in module tvm.relax.transform)
RemoveRedundantReshape (class in tvm.relax.transform)
RemoveStoreUndef() (in module tvm.tir.transform)
RemoveUnusedOutputs() (in module tvm.relax.transform)
RemoveUnusedParameters() (in module tvm.relax.transform)
RemoveWeightLayoutRewriteBlock() (in module tvm.tir.transform)
render() (tvm.instrument.PassTimingInstrument static method)
renew_defs() (in module tvm.tir.stmt_functor)
RenormalizeSplitPattern() (in module tvm.tir.transform)
reorder() (tvm.tir.schedule.Schedule method)
reorder_block_iter_var() (tvm.tir.schedule.Schedule method)
ReorderPermuteDimsAfterConcat() (in module tvm.relax.transform)
ReorderTakeAfterMatmul() (in module tvm.relax.transform)
repeat() (in module tvm.relax.op)
RepeatAttrs (class in tvm.relax.op.op_attrs)
replace() (tvm.tir.schedule.ScheduleState method)
replace_all_uses() (tvm.relax.DataflowBlockRewrite method)
replace_global_vars() (tvm.ir.IRModule method)
Report (class in tvm.runtime.profiling)
request() (tvm.rpc.TrackerSession method)
request_and_run() (tvm.rpc.TrackerSession method)
RequiredByConditional (tvm.tir.transform.HoistedLetBindings attribute)
reset_attr() (tvm.ir.Op method)
reshape() (in module tvm.relax.op)
resize1d() (in module tvm.topi.image)
resize2d() (in module tvm.relax.op.image)
(in module tvm.topi.image)
Resize2DAttrs (class in tvm.relax.op.op_attrs)
resize3d() (in module tvm.topi.image)
ret() (in module tvm.tir)
reverse_compute_at() (tvm.tir.schedule.Schedule method)
reverse_compute_inline() (tvm.tir.schedule.Schedule method)
RewriteCUDAGraph() (in module tvm.relax.transform)
RewriteDataflowReshape() (in module tvm.relax.transform)
RewriteUnsafeSelect() (in module tvm.tir.transform)
rfactor() (tvm.tir.schedule.Schedule method)
right_shift() (in module tvm.relax.op)
riscv_cpu() (in module tvm.target)
rms_norm() (in module tvm.relax.op.nn)
(in module tvm.topi.nn)
rocm() (in module tvm.runtime.ndarray)
(in module tvm.target)
(tvm.rpc.RPCSession method)
rocm_link() (in module tvm.contrib.rocm)
rolling_buffer() (tvm.tir.schedule.Schedule method)
root_mark() (tvm.tir.schedule.StmtSRef static method)
round() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
RPCError
RPCSession (class in tvm.rpc)
RPCSessionTimeoutError
rsqrt() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
run() (tvm.meta_schedule.Runner method)
run_after_pass() (tvm.instrument.PassInstrument method)
run_before_pass() (tvm.instrument.PassInstrument method)
RunCodegen() (in module tvm.relax.transform)
Runner (class in tvm.meta_schedule)
S
same_as() (tvm.ir.GlobalInfo method)
(tvm.ir.Type method)
(tvm.relax.StructInfo method)
(tvm.runtime.ndarray.NDArray method)
sample_categorical() (tvm.tir.schedule.Schedule method)
sample_compute_location() (tvm.tir.schedule.Schedule method)
sample_partitioned_tile() (tvm.tir.schedule.Schedule method)
sample_perfect_tile() (tvm.tir.schedule.Schedule method)
save() (tvm.meta_schedule.CostModel method)
save_function() (tvm.relax.VirtualMachine method)
(tvm.runtime.relax_vm.VirtualMachine method)
save_json() (in module tvm.ir)
scale_shift_nchw() (in module tvm.topi.nn)
scale_shift_nchwc() (in module tvm.topi.nn)
scale_shift_nhwc() (in module tvm.topi.nn)
scan() (in module tvm.te)
scan_axis (tvm.te.ScanOp property)
ScanOp (class in tvm.te)
ScanopAttrs (class in tvm.relax.op.op_attrs)
scatter() (tvm.runtime.disco.Session method)
scatter_elements() (in module tvm.relax.op)
scatter_from_worker0() (in module tvm.relax.op.ccl)
(tvm.runtime.disco.Session method)
scatter_nd() (in module tvm.relax.op)
Schedule (class in tvm.tir.schedule)
ScheduleDebugMask (class in tvm.tir.schedule)
ScheduleError
ScheduleRule (class in tvm.dlight)
(class in tvm.meta_schedule)
ScheduleState (class in tvm.tir.schedule)
scope() (tvm.tir.Buffer method)
SearchStrategy (class in tvm.meta_schedule)
seed() (tvm.tir.schedule.Schedule method)
Select (class in tvm.tir)
selu() (in module tvm.relax.op.nn)
SeqExpr (class in tvm.relax)
SeqStmt (class in tvm.tir)
Sequential (class in tvm.transform)
SequentialSpan (class in tvm.ir)
Server (class in tvm.rpc)
Session (class in tvm.runtime.disco)
set_attr() (tvm.ir.Op method)
set_attrs_type_key() (tvm.ir.Op method)
set_axis_separator() (tvm.tir.schedule.Schedule method)
set_input() (tvm.relax.VirtualMachine method)
(tvm.runtime.relax_vm.VirtualMachine method)
set_instrument() (tvm.relax.VirtualMachine method)
(tvm.runtime.relax_vm.VirtualMachine method)
set_keep_for_debug() (tvm.contrib.utils.TempDirectory class method)
set_num_evals() (tvm.transform.PassContext method)
set_num_inputs() (tvm.ir.Op method)
set_scope() (tvm.tir.schedule.Schedule method)
set_support_level() (tvm.ir.Op method)
set_var_remap() (tvm.relax.PyExprMutator method)
shape (tvm.te.Tensor property)
shape_of() (in module tvm.relax.op)
shape_to_tensor() (in module tvm.relax.op)
ShapeExpr (class in tvm.relax)
ShapeStructInfo (class in tvm.relax)
ShapeType (class in tvm.relax)
shift_left() (in module tvm.tir)
shift_right() (in module tvm.tir)
should_run() (tvm.instrument.PassInstrument method)
show() (tvm.tir.schedule.Schedule method)
(tvm.tir.schedule.Trace method)
Shuffle (class in tvm.tir)
shutdown() (tvm.runtime.disco.Session method)
sigmoid() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
sign() (in module tvm.relax.op)
silu() (in module tvm.relax.op.nn)
simdgroup_load() (in module tvm.tir)
simdgroup_multiply_accumulate() (in module tvm.tir)
simdgroup_store() (in module tvm.tir)
simplified() (tvm.tir.schedule.Trace method)
Simplify() (in module tvm.tir.transform)
simplify() (in module tvm.topi.image)
(in module tvm.topi.nn)
simulated_dequantize() (in module tvm.topi.nn)
simulated_quantize() (in module tvm.topi.nn)
sin() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
sinh() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
size_var() (in module tvm.te)
SizeVar (class in tvm.tir)
SkipAssert() (in module tvm.tir.transform)
SocketSession (class in tvm.runtime.disco)
softmax() (in module tvm.contrib.miopen)
(in module tvm.relax.op.nn)
(in module tvm.topi.nn)
softmax_common() (in module tvm.topi.nn)
SoftmaxAttrs (class in tvm.relax.op.op_attrs)
softplus() (in module tvm.relax.op.nn)
(in module tvm.topi.nn)
sort() (in module tvm.relax.op)
SortAttrs (class in tvm.relax.op.op_attrs)
SourceName (class in tvm.ir)
space_to_batch_nd() (in module tvm.topi.nn)
space_to_depth() (in module tvm.topi.nn)
SpaceGenerator (class in tvm.meta_schedule)
Span (class in tvm.ir)
specialize() (tvm.tir.PrimFunc method)
split() (in module tvm.relax.op)
(tvm.tir.schedule.Schedule method)
SplitAttrs (class in tvm.relax.op.op_attrs)
SplitCallTIRByPattern() (in module tvm.relax.transform)
SplitHostDevice() (in module tvm.tir.transform)
SplitLayoutRewritePreproc() (in module tvm.relax.transform)
sqrt() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
square() (in module tvm.relax.op)
squeeze() (in module tvm.relax.op)
SqueezeAttrs (class in tvm.relax.op.op_attrs)
start_checkpoint() (in module tvm.relax.op.grad)
start_profile_intrinsic() (in module tvm.tir)
state (tvm.tir.schedule.Schedule property)
StaticPlanBlockMemory() (in module tvm.relax.transform)
StatisticalAttrs (class in tvm.relax.op.op_attrs)
stats() (tvm.relax.VMExecutable method)
std() (in module tvm.relax.op)
stm32() (in module tvm.target)
Stmt (class in tvm.tir)
stmt (tvm.tir.schedule.StmtSRef property)
stmt_list() (in module tvm.tir)
stmt_seq() (in module tvm.tir)
StmtSRef (class in tvm.tir.schedule)
stop_lift_params() (in module tvm.relax.op.builtin)
storage_align() (tvm.tir.schedule.Schedule method)
StorageRewrite() (in module tvm.tir.transform)
stride_h (tvm.topi.nn.Workload attribute)
stride_w (tvm.topi.nn.Workload attribute)
strided_slice() (in module tvm.relax.op)
(in module tvm.topi.nn)
StridedSliceAttrs (class in tvm.relax.op.op_attrs)
StringImm (class in tvm.relax)
(class in tvm.tir)
struct_info (tvm.ir.RelaxExpr property)
struct_info_base_check() (in module tvm.relax.analysis)
struct_info_lca() (in module tvm.relax.analysis)
StructInfo (class in tvm.relax)
structural_equal() (in module tvm.ir)
structural_hash() (in module tvm.ir)
Sub (class in tvm.tir)
substitute() (in module tvm.tir.stmt_functor)
subtract() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
suggest_layout_transforms() (in module tvm.relax.analysis)
sum() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
summary() (tvm.rpc.TrackerSession method)
sync_worker_0() (tvm.runtime.disco.Session method)
system_lib() (tvm.rpc.RPCSession method)
T
table() (tvm.meta_schedule.Profiler method)
(tvm.runtime.profiling.Report method)
tag_scope() (in module tvm.te)
take() (in module tvm.relax.op)
take_backward() (in module tvm.relax.op.grad)
TakeAttrs (class in tvm.relax.op.op_attrs)
tan() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
tanh() (in module tvm.relax.op)
(in module tvm.te)
(in module tvm.tir)
tar() (in module tvm.contrib.tar)
Target (class in tvm.target)
target_or_current() (tvm.target.Target static method)
TargetKind (class in tvm.target)
TaskScheduler (class in tvm.meta_schedule)
tempdir() (in module tvm.contrib.utils)
TempDirectory (class in tvm.contrib.utils)
Tensor (class in tvm.te)
tensor_to_shape() (in module tvm.relax.op)
TensorIntrin (class in tvm.tir)
tensorize() (tvm.tir.schedule.Schedule method)
TensorSlice (class in tvm.te)
TensorStructInfo (class in tvm.relax)
TensorType (class in tvm.relax)
terminate() (tvm.rpc.Server method)
terminate_task() (tvm.meta_schedule.TaskScheduler method)
testing_scope() (tvm.relax.block_builder.BlockBuilder method)
TestingScope (class in tvm.relax.block_builder)
text_summary() (tvm.rpc.TrackerSession method)
thread_axis() (in module tvm.te)
thread_warp_size (tvm.target.Target property)
ThreadedSession (class in tvm.runtime.disco)
ThreadSync() (in module tvm.tir.transform)
tile() (in module tvm.relax.op)
TileAttrs (class in tvm.relax.op.op_attrs)
time_evaluator() (tvm.relax.VirtualMachine method)
(tvm.runtime.relax_vm.VirtualMachine method)
timeit() (tvm.meta_schedule.Profiler static method)
tir_vars_in_struct_info() (in module tvm.relax.analysis)
to_pytorch_func() (in module tvm.contrib.dlpack)
to_vdevice() (in module tvm.relax.op)
ToMixedPrecision() (in module tvm.relax.transform)
ToNonDataflow() (in module tvm.relax.transform)
topk() (in module tvm.relax.op)
TopKAttrs (class in tvm.relax.op.op_attrs)
TopologicalSort() (in module tvm.relax.transform)
touch_task() (tvm.meta_schedule.TaskScheduler method)
Trace (class in tvm.tir.schedule)
trace (tvm.tir.schedule.Schedule property)
trace() (in module tvm.te)
(in module tvm.tir)
TrackerSession (class in tvm.rpc)
transform_block_layout() (tvm.tir.schedule.Schedule method)
transform_layout() (tvm.tir.schedule.Schedule method)
TransformMmaBufferLayout() (in module tvm.tir.transform)
tril() (in module tvm.relax.op)
TriluAttrs (class in tvm.relax.op.op_attrs)
triu() (in module tvm.relax.op)
trunc() (in module tvm.te)
(in module tvm.tir)
truncdiv() (in module tvm.te)
(in module tvm.tir)
truncmod() (in module tvm.te)
(in module tvm.tir)
try_inline() (in module tvm.dlight)
try_inline_contiguous_spatial() (in module tvm.dlight)
tune() (tvm.meta_schedule.TaskScheduler method)
tune_tasks() (in module tvm.meta_schedule)
tune_tir() (in module tvm.meta_schedule)
TuneContext (class in tvm.meta_schedule)
Tuple (class in tvm.relax)
TupleGetItem (class in tvm.relax)
TupleStructInfo (class in tvm.relax)
TupleType (class in tvm.ir)
tvm.contrib
module
tvm.contrib.cblas
module
tvm.contrib.cc
module
tvm.contrib.clang
module
tvm.contrib.cublas
module
tvm.contrib.dlpack
module
tvm.contrib.emcc
module
tvm.contrib.miopen
module
tvm.contrib.ndk
module
tvm.contrib.nnpack
module
tvm.contrib.nvcc
module
tvm.contrib.pickle_memoize
module
tvm.contrib.random
module
tvm.contrib.rocblas
module
tvm.contrib.rocm
module
tvm.contrib.spirv
module
tvm.contrib.tar
module
tvm.contrib.utils
module
tvm.contrib.xcode
module
tvm.dlight
module
tvm.driver
module
tvm.error
module
tvm.instrument
module
tvm.ir
module
tvm.meta_schedule
module
tvm.relax
module
tvm.relax.analysis
module
tvm.relax.block_builder
module
tvm.relax.frontend
module
tvm.relax.frontend.onnx
module
tvm.relax.frontend.stablehlo
module
tvm.relax.frontend.torch
module
tvm.relax.op
module
tvm.relax.op.builtin
module
tvm.relax.op.ccl
module
tvm.relax.op.distributed
module
tvm.relax.op.grad
module
tvm.relax.op.image
module
tvm.relax.op.memory
module
tvm.relax.op.nn
module
tvm.relax.op.op_attrs
module
tvm.relax.transform
module
tvm.rpc
module
tvm.runtime
module
tvm.runtime.disco
module
tvm.runtime.ndarray
module
tvm.runtime.profiling
module
tvm.runtime.relax_vm
module
tvm.target
module
tvm.te
module
tvm.tir
module
tvm.tir.analysis.analysis
module
tvm.tir.schedule
module
tvm.tir.stmt_functor
module
tvm.tir.transform
module
tvm.topi.image
module
tvm.topi.nn
module
tvm.transform
module
tvm_access_ptr() (in module tvm.tir)
tvm_bmma_sync() (in module tvm.tir)
tvm_check_return() (in module tvm.tir)
tvm_fill_fragment() (in module tvm.tir)
tvm_load_matrix_sync() (in module tvm.tir)
tvm_mma_sync() (in module tvm.tir)
tvm_stack_alloca() (in module tvm.tir)
tvm_stack_make_array() (in module tvm.tir)
tvm_stack_make_shape() (in module tvm.tir)
tvm_store_matrix_sync() (in module tvm.tir)
tvm_struct_get() (in module tvm.tir)
tvm_struct_set() (in module tvm.tir)
tvm_thread_allreduce() (in module tvm.tir)
tvm_throw_last_error() (in module tvm.tir)
tvm_tuple() (in module tvm.tir)
TVMBackendAllocWorkspace() (in module tvm.tir)
TVMBackendFreeWorkspace() (in module tvm.tir)
TVMError
Type (class in tvm.ir)
type_annotation() (in module tvm.tir)
U
udchain() (in module tvm.relax.analysis)
unannotate() (tvm.tir.schedule.Schedule method)
undef() (in module tvm.tir)
undefined_vars() (in module tvm.tir.analysis.analysis)
uniform() (in module tvm.contrib.random)
UnifyThreadBinding() (in module tvm.tir.transform)
unique() (in module tvm.relax.op)
unpack_NCHWc_to_nchw() (in module tvm.topi.nn)
unroll() (tvm.tir.schedule.Schedule method)
UnrollLoop() (in module tvm.tir.transform)
unsafe_hide_buffer_access() (tvm.tir.schedule.Schedule method)
unsafe_set_dtype() (tvm.tir.schedule.Schedule method)
untar() (in module tvm.contrib.tar)
update() (tvm.ir.IRModule method)
(tvm.meta_schedule.CostModel method)
update_func() (tvm.ir.IRModule method)
(tvm.relax.block_builder.BlockBuilder method)
update_global_info() (tvm.ir.IRModule method)
UpdateParamStructInfo() (in module tvm.relax.transform)
UpdateVDevice() (in module tvm.relax.transform)
upload() (tvm.rpc.RPCSession method)
upsampling() (in module tvm.topi.nn)
upsampling3d() (in module tvm.topi.nn)
UseAssumeToReduceBranches() (in module tvm.tir.transform)
UsingBlockVar (tvm.tir.transform.HoistedConditionals attribute)
V
value_index (tvm.te.Tensor property)
Var (class in tvm.relax)
(class in tvm.tir)
var() (in module tvm.te)
VarBinding (class in tvm.relax)
variance() (in module tvm.relax.op)
VDevice (class in tvm.ir)
vectorcombine() (in module tvm.tir)
vectorhigh() (in module tvm.tir)
vectorize() (tvm.tir.schedule.Schedule method)
VectorizeLoop() (in module tvm.tir.transform)
vectorlow() (in module tvm.tir)
VERIFY_CACHED_FLAGS (tvm.tir.schedule.ScheduleDebugMask attribute)
verify_gpu_code() (in module tvm.tir.analysis.analysis)
verify_memory() (in module tvm.tir.analysis.analysis)
VERIFY_SREF_TREE (tvm.tir.schedule.ScheduleDebugMask attribute)
verify_ssa() (in module tvm.tir.analysis.analysis)
verify_well_formed() (in module tvm.tir.analysis.analysis)
VerifyMemory() (in module tvm.tir.transform)
VerifyVTCMLimit() (in module tvm.tir.transform)
view() (in module tvm.relax.op.memory)
VirtualDevice (class in tvm.target)
VirtualMachine (class in tvm.relax)
(class in tvm.runtime.relax_vm)
visit_binding() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_binding_block() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_binding_block_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_call_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_constant_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_data_type_imm_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_dataflow_block_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_dataflow_var_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_dataflow_var_def_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_expr() (tvm.relax.ExprFunctor method)
(tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_expr_post_order() (tvm.relax.PyExprMutator method)
visit_extern_func_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_function_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_global_var_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_if_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_match_cast_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_op_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_prim_value_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_seq_expr_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_shape_expr_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_span() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_string_imm_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_tuple_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_tuple_getitem_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_var_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_var_binding_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_var_def() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_var_def_() (tvm.relax.PyExprMutator method)
(tvm.relax.PyExprVisitor method)
visit_with_new_scope() (tvm.relax.PyExprMutator method)
vload() (tvm.tir.Buffer method)
VMBuiltinLower() (in module tvm.relax.transform)
VMExecutable (class in tvm.relax)
VMInstrumentReturnKind (class in tvm.relax)
(class in tvm.runtime.relax_vm)
VMShapeLower() (in module tvm.relax.transform)
vpi() (in module tvm.runtime.ndarray)
vscale() (in module tvm.tir)
vstore() (tvm.tir.Buffer method)
vulkan() (in module tvm.runtime.ndarray)
(tvm.rpc.RPCSession method)
W
WAR (tvm.tir.schedule.DepKind attribute)
WAW (tvm.tir.schedule.DepKind attribute)
webgpu() (in module tvm.runtime.ndarray)
(tvm.rpc.RPCSession method)
well_formed() (in module tvm.relax.analysis)
where() (in module tvm.relax.op)
which() (in module tvm.contrib.utils)
While (class in tvm.tir)
width (tvm.topi.nn.Workload attribute)
winograd_transform_matrices() (in module tvm.topi.nn)
with_attr() (tvm.ir.BaseFunc method)
(tvm.ir.IRModule method)
with_attrs() (tvm.ir.BaseFunc method)
(tvm.ir.IRModule method)
with_body() (tvm.tir.PrimFunc method)
with_decision() (tvm.tir.schedule.Trace method)
with_minrpc() (in module tvm.rpc)
with_struct_info() (tvm.relax.PyExprMutator method)
without_attr() (tvm.ir.BaseFunc method)
(tvm.ir.IRModule method)
work_on() (tvm.tir.schedule.Schedule method)
Workload (class in tvm.topi.nn)
wrap_param() (in module tvm.relax.op)
X
xcrun() (in module tvm.contrib.xcode)
Z
zeros() (in module tvm.relax.op)
zeros_like() (in module tvm.relax.op)