Index

A

abs() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) absolute() (tvm.testing.Path method) access_ptr() (tvm.tirx.Buffer method) accumulate (tvm.relax.op.op_attrs.IndexPutAttrs property) acos() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) acosh() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) 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) (in module tvm.topi.testing) adaptive_pool1d() (in module tvm.topi.nn) adaptive_pool3d() (in module tvm.topi.nn) AdaptivePool1DAttrs (class in tvm.relax.op.op_attrs) AdaptivePool2DAttrs (class in tvm.relax.op.op_attrs) AdaptivePool3DAttrs (class in tvm.relax.op.op_attrs) Add (class in tvm.tirx) add() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) (in module tvm.topi.nn) (tvm.relax.DataflowBlockRewrite method) add_argument() (tvm.ir.Op method) add_func() (tvm.relax.block_builder.BlockBuilder method) add_python_function() (tvm.relax.BasePyModule method) add_unit_loop() (tvm.s_tir.schedule.Schedule method) address_of() (in module tvm.tirx) AdjustMatmulOrder() (in module tvm.relax.transform) affine_grid() (in module tvm.relax.op.image) (in module tvm.topi.image) algo_to_index() (in module tvm.contrib.cudnn) aligned (tvm.relax.op.op_attrs.ROIAlignAttrs property) All (tvm.tirx.transform.HoistedConditionals attribute) (tvm.tirx.transform.HoistedLetBindings attribute) all() (in module tvm.te) (in module tvm.tirx) all_class_non_max_suppression() (in module tvm.relax.op) (in module tvm.relax.op.vision) all_global_vars() (in module tvm.relax.analysis) all_vars() (in module tvm.relax.analysis) AllClassNonMaximumSuppressionAttrs (class in tvm.relax.op.op_attrs) allgather() (in module tvm.relax.op.ccl) (tvm.runtime.disco.Session method) AllGatherAttrs (class in tvm.relax.op.op_attrs) alloc_storage() (in module tvm.relax.op.memory) (in module tvm.relax.op.vm) alloc_tensor() (in module tvm.relax.op.builtin) (in module tvm.relax.op.memory) (in module tvm.relax.op.vm) allocate_hexagon_array() (in module tvm.contrib.hexagon) AllocateWorkspace() (in module tvm.relax.transform) AllocBuffer (class in tvm.tirx) allreduce() (in module tvm.relax.op.ccl) (tvm.runtime.disco.Session method) AllReduceAttrs (class in tvm.relax.op.op_attrs) alpha (tvm.relax.op.op_attrs.LeakyReluAttrs property) AlterOpImpl() (in module tvm.relax.transform) Analyzer (class in tvm.arith) And (class in tvm.tirx) annotate() (tvm.s_tir.schedule.Schedule method) annotate_buffer_access() (tvm.s_tir.schedule.Schedule method) annotate_sharding() (in module tvm.relax.op.distributed) AnnotateEntryFunc() (in module tvm.tirx.transform) AnnotateTIROpPattern() (in module tvm.relax.transform) any() (in module tvm.te) (in module tvm.tirx) append() (tvm.s_tir.schedule.Trace method) append_or_update() (tvm.contrib.tvmjs.TensorCacheShardingManager method) AppendLoss() (in module tvm.relax.training) Apply() (in module tvm.tirx.transform) apply() (tvm.s_tir.dlight.ScheduleRule method) (tvm.s_tir.meta_schedule.MeasureCallback method) (tvm.s_tir.meta_schedule.Mutator method) (tvm.s_tir.meta_schedule.Postproc method) (tvm.s_tir.meta_schedule.ScheduleRule method) (tvm.tirx.Layout method) (tvm.tirx.Predicate method) apply_constraints_to_boolean_branches (tvm.tirx.transform.StmtSimplifyConfig property) apply_json_to_schedule() (tvm.s_tir.schedule.Trace static method) apply_to_schedule() (tvm.s_tir.schedule.Trace method)

apply_to_shape() (tvm.tirx.Layout method) ApplyDefaultSchedule (class in tvm.s_tir.dlight) arange() (in module tvm.relax.op) 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) Array (class in tvm.ir) as_json() (tvm.s_tir.schedule.Trace method) as_python() (tvm.relax.VMExecutable method) (tvm.s_tir.schedule.Trace method) as_text() (tvm.relax.VMExecutable method) asin() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) asinh() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) asobject() (tvm.te.TensorSlice method) assert_allclose() (in module tvm.testing) assert_op() (in module tvm.relax.op) assert_prim_expr_equal() (in module tvm.testing) assert_pure_function() (in module tvm.s_tir.analysis) assert_structural_equal() (in module tvm.ir) AssertStmt (class in tvm.tirx) assume() (in module tvm.tirx) assume_inbound (tvm.relax.op.op_attrs.StridedSliceAttrs property) 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.tirx) atan2() (in module tvm.relax.op) (in module tvm.tirx) atanh() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) 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_bias() (in module tvm.relax.op.nn) attention_python() (in module tvm.topi.testing) attention_var_len() (in module tvm.relax.op.nn) AttentionAttrs (class in tvm.relax.op.op_attrs) Attrs (class in tvm.ir) attrs (tvm.ir.BaseFunc property) (tvm.s_tir.schedule.Instruction attribute) AttrStmt (class in tvm.tirx) auto_max_depth (tvm.tirx.transform.UnrollLoopConfig property) auto_max_extent (tvm.tirx.transform.UnrollLoopConfig property) auto_max_step (tvm.tirx.transform.UnrollLoopConfig property) 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) axes (tvm.relax.op.op_attrs.GroupNormAttrs property) (tvm.relax.op.op_attrs.InstanceNormAttrs property) (tvm.relax.op.op_attrs.LayerNormAttrs property) (tvm.relax.op.op_attrs.PermuteDimsAttrs property) (tvm.relax.op.op_attrs.RMSNormAttrs property) axis (tvm.relax.op.op_attrs.ArgmaxArgminAttrs property) (tvm.relax.op.op_attrs.ArgsortAttrs property) (tvm.relax.op.op_attrs.BatchNormAttrs property) (tvm.relax.op.op_attrs.ConcatAttrs property) (tvm.relax.op.op_attrs.ExpandDimsAttrs property) (tvm.relax.op.op_attrs.FlipAttrs property) (tvm.relax.op.op_attrs.GatherElementsAttrs property) (tvm.relax.op.op_attrs.OneHotAttrs property) (tvm.relax.op.op_attrs.PReluAttrs property) (tvm.relax.op.op_attrs.QuantizeAttrs property) (tvm.relax.op.op_attrs.RepeatAttrs property) (tvm.relax.op.op_attrs.ScanopAttrs property) (tvm.relax.op.op_attrs.ScatterCollectiveAttrs property) (tvm.relax.op.op_attrs.ScatterElementsAttrs property) (tvm.relax.op.op_attrs.SliceScatterAttrs property) (tvm.relax.op.op_attrs.SoftmaxAttrs property) (tvm.relax.op.op_attrs.SortAttrs property) (tvm.relax.op.op_attrs.SplitAttrs property) (tvm.relax.op.op_attrs.SqueezeAttrs property) (tvm.relax.op.op_attrs.StackAttrs property) (tvm.relax.op.op_attrs.StatisticalAttrs property) (tvm.relax.op.op_attrs.TakeAttrs property) (tvm.relax.op.op_attrs.TopKAttrs property) axis_separators (tvm.relax.op.op_attrs.LayoutTransformAttrs property)

B

BaseCheckResult (class in tvm.relax.analysis) BaseExpr (class in tvm.ir) BaseFunc (class in tvm.ir) BasePyModule (class in tvm.relax) batch_dims (tvm.relax.op.op_attrs.GatherNDAttrs property) batch_flatten() (in module tvm.relax.op.nn) batch_matmul() (in module tvm.contrib.cblas) (in module tvm.contrib.cublas) (in module tvm.contrib.hipblas) (in module tvm.contrib.mkl) (in module tvm.topi.nn) (in module tvm.topi.testing) batch_norm() (in module tvm.relax.op.nn) (in module tvm.topi.nn) (in module tvm.topi.testing) batch_to_space_nd() (in module tvm.topi.nn) batch_to_space_nd_python() (in module tvm.topi.testing) BatchNormAttrs (class in tvm.relax.op.op_attrs) begin_scope() (tvm.relax.block_builder.BlockBuilder method) beta (tvm.relax.op.op_attrs.SoftplusAttrs property) BF16ComputeLegalize() (in module tvm.tirx.transform) BF16StorageLegalize() (in module tvm.tirx.transform) binarize_pack() (in module tvm.topi.nn) binary_dense() (in module tvm.topi.nn) Bind (class in tvm.tirx) (tvm.tirx.transform.HoistedLetBindings attribute) bind() (tvm.arith.Analyzer method) (tvm.s_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.tirx.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.backend.cuda.op) (in module tvm.relax.op) (in module tvm.tirx) bitwise_not() (in module tvm.relax.op) (in module tvm.tirx) bitwise_or() (in module tvm.relax.op) (in module tvm.tirx) bitwise_xor() (in module tvm.relax.op) (in module tvm.tirx) BlockBuilder (class in tvm.relax.block_builder) blockize() (tvm.s_tir.schedule.Schedule method) BooleanExpression (tvm.tirx.transform.HoistedConditionals attribute) bound_vars() (in module tvm.relax.analysis) break_loop() (in module tvm.tirx) Broadcast (class in tvm.tirx) broadcast() (tvm.runtime.disco.Session method) (tvm.tirx.Layout method) broadcast_from_worker0() (in module tvm.relax.op.ccl) (tvm.runtime.disco.Session method) broadcast_to() (in module tvm.relax.op) bucketize() (in module tvm.relax.op) Buffer (class in tvm.tirx) BufferLoad (class in tvm.tirx) BufferRegion (class in tvm.tirx) BufferStore (class in tvm.tirx) build() (in module tvm.relax) (in module tvm.tirx) (tvm.s_tir.meta_schedule.Builder method) Builder (class in tvm.s_tir.meta_schedule) BundleModelParams() (in module tvm.relax.transform) byte_offset (tvm.tirx.Buffer property) byte_offset_of() (tvm.tirx.Buffer method)

C

c() (tvm.relax.ExecBuilder method) Cache (class in tvm.contrib.pickle_memoize) cache (tvm.contrib.pickle_memoize.Cache property) cache_index() (tvm.s_tir.schedule.Schedule method) cache_inplace() (tvm.s_tir.schedule.Schedule method) cache_read() (tvm.s_tir.schedule.Schedule method) cache_write() (tvm.s_tir.schedule.Schedule method) calculate_allocated_bytes() (in module tvm.s_tir.analysis) Call (class in tvm.relax) call_builtin_with_ctx() (in module tvm.relax.op) call_cpacked() (in module tvm.tirx) call_cpacked_lowered() (in module tvm.tirx) call_dps_packed() (in module tvm.relax) (in module tvm.relax.op) (tvm.relax.BasePyModule method) call_extern() (in module tvm.tirx) call_inplace_packed() (in module tvm.relax.op) call_intrin() (in module tvm.backend.cuda.op) (in module tvm.tirx) call_llvm_intrin() (in module tvm.tirx) call_llvm_pure_intrin() (in module tvm.tirx) call_packed() (in module tvm.tirx) (tvm.runtime.disco.Session method) call_packed_lowered() (in module tvm.tirx) call_pure_extern() (in module tvm.tirx) call_pure_packed() (in module tvm.relax) (in module tvm.relax.op) call_py_func() (in module tvm.relax.op) (tvm.relax.BasePyModule method) 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.tirx) (tvm.relax.BasePyModule method) call_tir_dyn() (in module tvm.relax.op.vm) 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.tirx) CallingConv (class in tvm.ir) CallInplacePackedAttrs (class in tvm.relax.op.op_attrs) CallTIRInplaceAttrs (class in tvm.relax.op.op_attrs) 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.s_tir.schedule.Schedule method) can_prove() (tvm.arith.Analyzer method) can_prove_equal() (tvm.arith.Analyzer method) canonical_simplify() (tvm.arith.Analyzer method) canonicalize() (tvm.tirx.Layout method) CanonicalizeBindings() (in module tvm.relax.transform) Cast (class in tvm.tirx) causal_mask (tvm.relax.op.op_attrs.AttentionAttrs property) ceil() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) ceil_mode (tvm.relax.op.op_attrs.Pool1DAttrs property) (tvm.relax.op.op_attrs.Pool2DAttrs property) (tvm.relax.op.op_attrs.Pool3DAttrs property) ceildiv() (in module tvm.tirx) center (tvm.relax.op.op_attrs.BatchNormAttrs property) (tvm.relax.op.op_attrs.GroupNormAttrs property) (tvm.relax.op.op_attrs.InstanceNormAttrs property) (tvm.relax.op.op_attrs.LayerNormAttrs property) channel_axis (tvm.relax.op.op_attrs.GroupNormAttrs property) (tvm.relax.op.op_attrs.InstanceNormAttrs property) check_bool_expr_is_true() (in module tvm.testing) check_int_constraints_trans_consistency() (in module tvm.testing) check_numerical_grads() (in module tvm.testing) check_well_formed() (in module tvm.relax.analysis) chmod() (tvm.testing.Path method) circular_pad() (in module tvm.topi.nn) cl() (tvm.rpc.RPCSession method) clip (tvm.relax.op.op_attrs.MultiboxTransformLocAttrs property) clip() (in module tvm.relax.op) clone() (tvm.arith.Analyzer method) (tvm.s_tir.meta_schedule.Mutator method) (tvm.s_tir.meta_schedule.Postproc method) (tvm.s_tir.meta_schedule.ScheduleRule method) (tvm.s_tir.meta_schedule.SearchStrategy method) (tvm.s_tir.meta_schedule.SpaceGenerator method) (tvm.s_tir.meta_schedule.TuneContext method) close() (tvm.contrib.hexagon.ContainerSession method) (tvm.rpc.TrackerSession method) clz() (in module tvm.tirx) 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) CombineParallelMatmul() (in module tvm.relax.transform) comm_reducer() (in module tvm.te) (in module tvm.tirx) commit() (tvm.contrib.tvmjs.TensorCacheShardingManager method) commit_tuning_record() (tvm.s_tir.meta_schedule.Database method) commit_workload() (tvm.s_tir.meta_schedule.Database method) CommonSubexprElim() (in module tvm.tirx.transform) CommReducer (class in tvm.tirx) CompareResult (class in tvm.arith) compile() (in module tvm) compile_coreml() (in module tvm.support.xcode) compile_cuda() (in module tvm.support.nvcc) compile_metal() (in module tvm.support.xcode) ComposeLayout (class in tvm.tirx) computable_at_compile_time() (in module tvm.relax.analysis) compute() (in module tvm.te) compute_at() (tvm.s_tir.schedule.Schedule method) compute_inline() (tvm.s_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.backend.cuda.op) (in module tvm.relax) (in module tvm.te) const_int_bound() (tvm.arith.Analyzer method) const_int_bound_is_bound() (tvm.arith.Analyzer method) Constant (class in tvm.relax) ConstIntBound (class in tvm.arith) constraint_scope() (tvm.arith.Analyzer method) ContainerSession (class in tvm.contrib.hexagon) contains_impure_call() (in module tvm.relax.analysis) continue_loop() (in module tvm.tirx) 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_ncw_python() (in module tvm.topi.testing) conv1d_nwc() (in module tvm.topi.nn) conv1d_transpose() (in module tvm.relax.op.nn) conv1d_transpose_ncw() (in module tvm.topi.nn) conv1d_transpose_ncw_python() (in module tvm.topi.testing) Conv1DAttrs (class in tvm.relax.op.op_attrs) Conv1DTransposeAttrs (class in tvm.relax.op.op_attrs) conv2d() (in module tvm.relax.op.nn) (in module tvm.topi.nn) conv2d_backward_weight_python() (in module tvm.topi.testing) conv2d_hwcn() (in module tvm.topi.nn) conv2d_hwcn_python() (in module tvm.topi.testing) conv2d_nchw() (in module tvm.topi.nn) conv2d_nchw_python() (in module tvm.topi.testing) conv2d_NCHWc() (in module tvm.topi.nn) conv2d_NCHWc_int8() (in module tvm.topi.nn) conv2d_NCHWc_OIHWo() (in module tvm.topi.nn)

conv2d_nhwc() (in module tvm.topi.nn) conv2d_nhwc_python() (in module tvm.topi.testing) 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_transpose_nchw_python() (in module tvm.topi.testing) conv2d_transpose_nhwc_python() (in module tvm.topi.testing) 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_ncdhw_python() (in module tvm.topi.testing) conv3d_ndhwc() (in module tvm.topi.nn) conv3d_ndhwc_python() (in module tvm.topi.testing) conv3d_transpose() (in module tvm.relax.op.nn) conv3d_transpose_ncdhw() (in module tvm.topi.nn) conv3d_transpose_ncdhw_preprocess() (in module tvm.topi.nn) conv3d_transpose_ncdhw_python() (in module tvm.topi.testing) conv3d_winograd_weight_transform() (in module tvm.topi.nn) Conv3DAttrs (class in tvm.relax.op.op_attrs) Conv3DTransposeAttrs (class in tvm.relax.op.op_attrs) conv_backward_data() (in module tvm.contrib.cudnn) conv_backward_data_find_algo() (in module tvm.contrib.cudnn) conv_backward_filter() (in module tvm.contrib.cudnn) conv_backward_filter_find_algo() (in module tvm.contrib.cudnn) conv_dgrad_shape() (in module tvm.contrib.cudnn) conv_forward() (in module tvm.contrib.cudnn) conv_forward_find_algo() (in module tvm.contrib.cudnn) conv_output_shape() (in module tvm.contrib.cudnn) convert_boolean_to_and_of_ors (tvm.tirx.transform.StmtSimplifyConfig property) convert_func() (in module tvm.contrib.dlpack) convert_to_expr() (in module tvm.relax) ConvertLayout() (in module tvm.relax.transform) ConvertSSA() (in module tvm.tirx.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) coord_start (tvm.relax.op.op_attrs.NonMaximumSuppressionAttrs property) coordinate_transformation_mode (tvm.relax.op.op_attrs.Resize2DAttrs property) copy() (tvm.s_tir.schedule.Schedule method) copy_from() (tvm.contrib.hexagon.ContainerSession method) copy_from_worker_0() (tvm.runtime.disco.Session method) copy_to() (tvm.contrib.hexagon.ContainerSession method) copy_to_worker_0() (tvm.runtime.disco.Session method) copysign() (in module tvm.tirx) CoreMLModule (class in tvm.contrib.coreml_runtime) correlation_nchw() (in module tvm.topi.nn) correlation_nchw_python() (in module tvm.topi.testing) cos() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) cosh() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) CostModel (class in tvm.s_tir.meta_schedule) count() (tvm.topi.nn.Sequence method) count_include_pad (tvm.relax.op.op_attrs.Pool1DAttrs property) (tvm.relax.op.op_attrs.Pool2DAttrs property) (tvm.relax.op.op_attrs.Pool3DAttrs property) cpu() (tvm.rpc.RPCSession method) create() (in module tvm.contrib.coreml_runtime) (tvm.s_tir.meta_schedule.Builder static method) (tvm.s_tir.meta_schedule.CostModel static method) (tvm.s_tir.meta_schedule.Database static method) (tvm.s_tir.meta_schedule.FeatureExtractor static method) (tvm.s_tir.meta_schedule.MeasureCallback static method) (tvm.s_tir.meta_schedule.Mutator static method) (tvm.s_tir.meta_schedule.Postproc static method) (tvm.s_tir.meta_schedule.Runner static method) (tvm.s_tir.meta_schedule.ScheduleRule static method) (tvm.s_tir.meta_schedule.SearchStrategy static method) (tvm.s_tir.meta_schedule.SpaceGenerator static method) (tvm.s_tir.meta_schedule.TaskScheduler static method) create_aot_shared() (in module tvm.contrib.hexagon) create_dylib() (in module tvm.support.xcode) create_empty() (tvm.relax.Function static method) create_executable() (in module tvm.support.cc) create_llvm() (in module tvm.support.clang) create_prim_func() (in module tvm.te) create_shared() (in module tvm.contrib.hexagon) (in module tvm.support.cc) (in module tvm.support.ndk) create_staticlib() (in module tvm.support.cc) (in module tvm.support.ndk) create_tvmjs_wasm() (in module tvm.support.emcc) crop_and_resize() (in module tvm.topi.image) crop_and_resize_python() (in module tvm.topi.testing) cross_compiler() (in module tvm.support.cc) cross_entropy_with_logits() (in module tvm.relax.op.nn) cubic_alpha (tvm.relax.op.op_attrs.Resize2DAttrs property) cubic_exclude (tvm.relax.op.op_attrs.Resize2DAttrs property) cuda() (tvm.rpc.RPCSession method) cuda_atomic_add() (in module tvm.backend.cuda.op) cuda_atomic_cas() (in module tvm.backend.cuda.op) cuda_bfloat162float() (in module tvm.backend.cuda.op) cuda_cluster_sync() (in module tvm.backend.cuda.op) cuda_copy_128b() (in module tvm.backend.cuda.op) cuda_copy_16b() (in module tvm.backend.cuda.op) cuda_copy_32b() (in module tvm.backend.cuda.op) cuda_copy_64b() (in module tvm.backend.cuda.op) cuda_copy_8b() (in module tvm.backend.cuda.op) cuda_copy_bytes() (in module tvm.backend.cuda.op) cuda_cta_max() (in module tvm.backend.cuda.op) cuda_cta_min() (in module tvm.backend.cuda.op) cuda_cta_reduce() (in module tvm.backend.cuda.op) cuda_cta_sum() (in module tvm.backend.cuda.op) cuda_cta_sync() (in module tvm.backend.cuda.op) cuda_cvta_generic_to_shared() (in module tvm.backend.cuda.op) cuda_float22half2() (in module tvm.backend.cuda.op) cuda_float8tohalf8() (in module tvm.backend.cuda.op) cuda_func_call() (in module tvm.backend.cuda.op) cuda_get_tmem_addr() (in module tvm.backend.cuda.op) cuda_grid_sync() (in module tvm.backend.cuda.op) cuda_half2float() (in module tvm.backend.cuda.op) cuda_half8tofloat8() (in module tvm.backend.cuda.op) cuda_ldg() (in module tvm.backend.cuda.op) cuda_nano_sleep() (in module tvm.backend.cuda.op) cuda_printf() (in module tvm.backend.cuda.op) cuda_runtime_instr_desc() (in module tvm.backend.cuda.op) cuda_sm100_tma_2sm_mbarrier_addr() (in module tvm.backend.cuda.op) cuda_smem_addr_from_uint64() (in module tvm.backend.cuda.op) cuda_syncthreads_and() (in module tvm.backend.cuda.op) cuda_syncthreads_or() (in module tvm.backend.cuda.op) cuda_thread_fence() (in module tvm.backend.cuda.op) cuda_thread_rank() (in module tvm.backend.cuda.op) cuda_trap_when_assert_failed() (in module tvm.backend.cuda.op) cuda_warp_max() (in module tvm.backend.cuda.op) cuda_warp_min() (in module tvm.backend.cuda.op) cuda_warp_reduce() (in module tvm.backend.cuda.op) cuda_warp_sum() (in module tvm.backend.cuda.op) cuda_warp_sync() (in module tvm.backend.cuda.op) cuda_warpgroup_sync() (in module tvm.backend.cuda.op) CUDANamespace (class in tvm.backend.cuda.script) cumprod() (in module tvm.relax.op) cumsum() (in module tvm.relax.op) current() (tvm.ir.transform.PassContext static method) (tvm.relax.block_builder.BlockBuilder static method) (tvm.s_tir.meta_schedule.Database static method) (tvm.s_tir.meta_schedule.Profiler static method) (tvm.script.ir_builder.IRBuilder static method) (tvm.target.Target static method) current_block_is_dataflow() (tvm.relax.block_builder.BlockBuilder method) cwd() (tvm.testing.Path class method)

D

data_layout (tvm.relax.op.op_attrs.Conv1DAttrs property) (tvm.relax.op.op_attrs.Conv1DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv2DAttrs property) (tvm.relax.op.op_attrs.Conv2DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv3DAttrs property) (tvm.relax.op.op_attrs.Conv3DTransposeAttrs property) Database (class in tvm.s_tir.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.tirx) 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.s_tir.schedule.Trace attribute) decl_function() (in module tvm.script.ir_builder.ir) 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.tirx) decompose_padding() (tvm.s_tir.schedule.Schedule method) decompose_reduction() (tvm.s_tir.schedule.Schedule method) DecomposeOpsForInference() (in module tvm.relax.transform) DecomposeOpsForTraining() (in module tvm.relax.transform) deduce_bound() (in module tvm.arith) def_function() (in module tvm.script.ir_builder.ir) default_tir_pipeline() (in module tvm.s_tir.backend.adreno.pipeline) definable_tir_vars_in_type() (in module tvm.relax.analysis) defined_symbolic_vars() (in module tvm.relax.analysis) deformable_conv2d_nchw() (in module tvm.topi.nn) deformable_conv2d_nchw_python() (in module tvm.topi.testing) deformable_conv2d_nhwc() (in module tvm.topi.nn) deformable_conv2d_nhwc_python() (in module tvm.topi.testing) dense() (in module tvm.topi.nn) (in module tvm.topi.testing) dense_pack() (in module tvm.topi.nn) Dependency (class in tvm.s_tir.schedule) DepKind (class in tvm.s_tir.schedule) depth (tvm.relax.op.op_attrs.OneHotAttrs property) depth_to_space() (in module tvm.topi.nn) depth_to_space_python() (in module tvm.topi.testing) 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) depthwise_conv2d_python_nchw() (in module tvm.topi.testing) depthwise_conv2d_python_nchwc() (in module tvm.topi.testing) depthwise_conv2d_python_nhwc() (in module tvm.topi.testing) dequantize() (in module tvm.relax.op) derive_call_ret_type() (in module tvm.relax.analysis) descending (tvm.relax.op.op_attrs.ArgsortAttrs property) (tvm.relax.op.op_attrs.SortAttrs property) describe() (in module tvm.support) detach_params() (in module tvm.relax.frontend) detect_buffer_access_lca() (in module tvm.s_tir.analysis) detect_clip_bound() (in module tvm.arith) detect_iter_map() (in module tvm.arith)

detect_linear_equation() (in module tvm.arith) detect_recursion() (in module tvm.relax.analysis) device() (tvm.rpc.RPCSession method) device_enabled() (in module tvm.testing) device_mesh() (in module tvm.relax.distributed) device_type (tvm.relax.op.op_attrs.HintOnDeviceAttrs property) device_type_int (tvm.target.VirtualDevice property) DeviceMesh (class in tvm.relax.distributed) DiagnosticError DictAttrs (class in tvm.ir) dilate() (in module tvm.topi.nn) dilate_python() (in module tvm.topi.testing) dilation (tvm.relax.op.op_attrs.Conv1DAttrs property) (tvm.relax.op.op_attrs.Conv1DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv2DAttrs property) (tvm.relax.op.op_attrs.Conv2DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv3DAttrs property) (tvm.relax.op.op_attrs.Conv3DTransposeAttrs property) (tvm.relax.op.op_attrs.Pool1DAttrs property) (tvm.relax.op.op_attrs.Pool2DAttrs property) (tvm.relax.op.op_attrs.Pool3DAttrs property) 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) direct_sum() (tvm.tirx.Layout method) DirectoryCreatedPastAtExit DispatchSampling (class in tvm.relax.backend) DispatchSortScan (class in tvm.relax.backend) Div (class in tvm.tirx) div() (in module tvm.te) (in module tvm.tirx) divide() (in module tvm.relax.op) DModule (class in tvm.runtime.disco) dnnl_conv2d() (in module tvm.contrib.dnnl) dom (tvm.s_tir.dlight.IterInfo property) dom() (tvm.s_tir.dlight.SBlockInfo method) dom_kind() (tvm.s_tir.dlight.SBlockInfo method) download() (in module tvm.contrib.download) (tvm.rpc.RPCSession method) download_linked_module() (tvm.rpc.RPCSession method) download_testdata() (in module tvm.contrib.download) dp4a() (in module tvm.tirx) 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) dst_vdevice (tvm.relax.op.op_attrs.ToVDeviceAttrs property) DTensorType (class in tvm.relax.distributed) dtype (tvm.relax.op.op_attrs.ArgsortAttrs property) (tvm.relax.op.op_attrs.AstypeAttrs property) (tvm.relax.op.op_attrs.InitAttrs property) (tvm.relax.op.op_attrs.MultinomialFromUniformAttrs property) (tvm.relax.op.op_attrs.ScanopAttrs property) (tvm.relax.op.op_attrs.TopKAttrs property) (tvm.relax.op.op_attrs.WrapParamAttrs property) (tvm.te.TensorSlice property) dummy_global_info() (in module tvm.script.ir_builder.ir) DummyGlobalInfo (class in tvm.ir) dump_pruned() (tvm.s_tir.meta_schedule.Database method) dump_tensor_cache() (in module tvm.contrib.tvmjs) DumpIR (class in tvm.ir.instrument) 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) elem_offset_of() (tvm.tirx.Buffer method) 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() (tvm.runtime.disco.Session method) enabled() (in module tvm.s_tir.tensor_intrin) enabled_extensions (tvm.arith.Analyzer property) enabled_targets() (in module tvm.testing) end_checkpoint() (in module tvm.relax.op.grad) end_profile_intrinsic() (in module tvm.tirx) end_scope() (tvm.relax.block_builder.BlockBuilder method) ensure_zero_offset() (in module tvm.relax.op.memory) enter_pass_ctx() (tvm.ir.instrument.PassInstrument method) enter_postproc() (tvm.s_tir.schedule.Schedule method) EnvFunc (class in tvm.ir) epsilon (tvm.relax.op.op_attrs.BatchNormAttrs property) (tvm.relax.op.op_attrs.GroupNormAttrs property) (tvm.relax.op.op_attrs.InstanceNormAttrs property) (tvm.relax.op.op_attrs.LayerNormAttrs property) (tvm.relax.op.op_attrs.RMSNormAttrs property) EQ (class in tvm.tirx) 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.tirx) estimate_memory_usage() (in module tvm.relax.analysis) estimate_region_lower_bound() (in module tvm.arith) estimate_region_strict_bound() (in module tvm.arith) estimate_region_upper_bound() (in module tvm.arith) estimate_tir_flops() (in module tvm.s_tir.analysis) Evaluate (class in tvm.tirx)

ewise_fma() (in module tvm.relax.op) exclusive (tvm.relax.op.op_attrs.ScanopAttrs property) exec() (tvm.contrib.hexagon.ContainerSession method) ExecBuilder (class in tvm.relax) ExecScope (class in tvm.tirx) exists() (in module tvm.contrib.cudnn) (tvm.testing.Path method) exit_pass_ctx() (tvm.ir.instrument.PassInstrument method) exp() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) exp10() (in module tvm.tirx) exp2() (in module tvm.tirx) 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) expanduser() (tvm.testing.Path method) explicit_unroll (tvm.tirx.transform.UnrollLoopConfig property) export_module() (in module tvm.contrib.hexagon) export_params() (tvm.relax.training.Trainer method) export_runtime() (in module tvm.contrib.tvmjs) export_states() (tvm.relax.training.Trainer method) Expr (in module tvm.relax) (in module tvm.relax.op.grad) (in module tvm.relax.op.vision) expr_deep_equal() (in module tvm.tirx.analysis.analysis) ExprFunctor (class in tvm.relax) (class in tvm.tirx) ExprRV (in module tvm.s_tir.schedule) ext_dev() (tvm.rpc.RPCSession method) Extension (class in tvm.arith) 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.s_tir.meta_schedule.FeatureExtractor method) ExtractedTask (class in tvm.s_tir.meta_schedule) extrapolation_value (tvm.relax.op.op_attrs.Resize2DAttrs property) eye() (in module tvm.relax.op) eye_like() (in module tvm.relax.op)

F

f() (tvm.relax.ExecBuilder method) fast_softmax() (in module tvm.topi.nn) FastMathTransform (class in tvm.relax.transform) FeatureExtractor (class in tvm.s_tir.meta_schedule) fifo_buffer() (in module tvm.topi.nn) FileLock (class in tvm.support.utils) filelock() (in module tvm.support.utils) Filter() (in module tvm.tirx.transform) finalize() (tvm.relax.block_builder.BlockBuilder method) finalize_modules() (in module tvm.contrib.cutlass) find_anchor_sblock() (in module tvm.s_tir.analysis) find_clang() (in module tvm.support.clang) find_cuda_path() (in module tvm.support.nvcc) find_lld() (in module tvm.support.rocm) find_nvshmem_paths() (in module tvm.support.nvcc) find_rocm_path() (in module tvm.support.rocm) find_wasm_lib() (in module tvm.support.emcc) finish() (tvm.contrib.tvmjs.TensorCacheShardingManager method) fixture() (in module tvm.testing) flatten() (in module tvm.relax.op) (in module tvm.topi.nn) FlattenBuffer() (in module tvm.tirx.transform) flip() (in module tvm.relax.op) FlipAttrs (class in tvm.relax.op.op_attrs) FloatImm (class in tvm.tirx) floor() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) floor_divide() (in module tvm.relax.op) floor_mod() (in module tvm.relax.op) FloorDiv (class in tvm.tirx) floordiv() (in module tvm.te) (in module tvm.tirx) FloorMod (class in tvm.tirx) floormod() (in module tvm.te) (in module tvm.tirx) fmod() (in module tvm.te) (in module tvm.tirx) FoldBatchnormToConv2D (class in tvm.relax.transform) FoldConstant() (in module tvm.relax.transform)

For (class in tvm.tirx) force_suppress (tvm.relax.op.op_attrs.NonMaximumSuppressionAttrs property) ForceNarrowIndexToInt32() (in module tvm.tirx.transform) fork_seed() (tvm.s_tir.schedule.Schedule method) ForKind (class in tvm.tirx) FP8ComputeLegalize() (in module tvm.tirx.transform) FP8StorageLegalize() (in module tvm.tirx.transform) free_symbolic_vars() (in module tvm.relax.analysis) free_vars() (in module tvm.relax.analysis) from_callable() (tvm.s_tir.dlight.ScheduleRule static method) from_device() (tvm.target.Target static method) from_exported_program() (in module tvm.relax.frontend.torch) from_expr() (tvm.ir.IRModule static method) from_func() (tvm.tirx.IndexMap static method) from_func_with_separators() (tvm.tirx.IndexMap static method) from_fx() (in module tvm.relax.frontend.torch) from_iters() (tvm.tirx.TileLayout static 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) from_text() (tvm.relax.distributed.Placement static method) from_tflite() (in module tvm.relax.frontend.tflite) 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.s_tir.schedule.Schedule property) 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.s_tir.schedule.Schedule method) fuse_reduction_epilogue() (tvm.s_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_nd_python() (in module tvm.topi.testing) gather_python() (in module tvm.topi.testing) gather_to_worker0() (tvm.runtime.disco.Session method) GatherElementsAttrs (class in tvm.relax.op.op_attrs) GatherNDAttrs (class in tvm.relax.op.op_attrs) GE (class in tvm.tirx) gelu() (in module tvm.relax.op.nn) gelu_tanh() (in module tvm.relax.op.nn) generate_design_space() (tvm.s_tir.meta_schedule.SpaceGenerator method) (tvm.s_tir.meta_schedule.TuneContext method) generate_measure_candidates() (tvm.s_tir.meta_schedule.SearchStrategy method) (tvm.s_tir.meta_schedule.TuneContext method) generate_random_array() (in module tvm.testing) get() (tvm.ir.DictAttrs method) (tvm.ir.EnvFunc static method) (tvm.ir.Map method) (tvm.ir.Op static method) (tvm.relax.block_builder.BlockBuilder method) (tvm.relax.ExecBuilder method) (tvm.s_tir.meta_schedule.Profiler method) (tvm.s_tir.schedule.InstructionKind static method) (tvm.s_tir.schedule.Schedule method) (tvm.script.ir_builder.IRBuilder method) (tvm.tirx.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.tirx) get_all_tuning_records() (tvm.s_tir.meta_schedule.Database method) get_attr() (tvm.ir.IRModule method) (tvm.ir.Op method) get_cc() (in module tvm.support.cc) get_child_blocks() (tvm.s_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.s_tir.schedule.Schedule method) get_cuda_version() (in module tvm.support.nvcc) get_decision() (tvm.s_tir.schedule.Trace method) get_default_pipeline() (in module tvm.relax) get_default_tir_pipeline() (in module tvm.tirx) get_deps_by_dst() (tvm.s_tir.schedule.SBlockScope method) get_deps_by_src() (tvm.s_tir.schedule.SBlockScope method) get_env() (tvm.contrib.hexagon.ContainerSession method) get_flattened_buffer() (tvm.tirx.Buffer method) get_function() (tvm.relax.BasePyModule method) (tvm.rpc.RPCSession method) get_global_func() (tvm.runtime.disco.Session method) get_global_symbol_section_map() (in module tvm.support.cc) (in module tvm.support.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.s_tir.schedule.Schedule method) get_output_blocks() (tvm.s_tir.schedule.Schedule method) get_outputs() (tvm.relax.VirtualMachine method) (tvm.runtime.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_padded_shape() (in module tvm.topi.nn) get_pattern() (in module tvm.relax.backend) get_patterns_with_prefix() (in module tvm.relax.backend) get_pipeline() (in module tvm.relax) get_private_buffers() (tvm.tirx.TilePrimitiveCall method) get_producers() (tvm.s_tir.schedule.Schedule method) get_sblock() (tvm.s_tir.schedule.Schedule method) get_sblock_access_region() (in module tvm.s_tir.analysis) get_sblock_read_write_region() (in module tvm.s_tir.analysis) get_sblock_scope() (tvm.s_tir.schedule.ScheduleState method) get_scope() (tvm.tirx.TileLayout method) get_shape_of() (in module tvm.relax) get_smtlib2() (tvm.arith.Analyzer method) get_sref() (tvm.s_tir.schedule.Schedule method) (tvm.s_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.support.cc) get_target_device_type() (tvm.target.Target method) get_tir_pipeline() (in module tvm.tirx) get_top_k() (tvm.s_tir.meta_schedule.Database method) get_unique_name() (tvm.relax.block_builder.BlockBuilder method) get_valid_counts() (in module tvm.relax.op) (in module tvm.relax.op.vision) get_valid_counts_python() (in module tvm.topi.testing) get_var2val() (in module tvm.relax.analysis) get_var_remap() (tvm.relax.PyExprMutator method) get_vscale_expr() (in module tvm.tirx) get_vtcm_compaction_passes() (in module tvm.s_tir.analysis) get_z3_stats() (tvm.arith.Analyzer method) GetValidCountsAttrs (class in tvm.relax.op.op_attrs) glob() (tvm.testing.Path method) global_pool() (in module tvm.topi.nn) GlobalInfo (class in tvm.ir) GlobalVar (class in tvm.ir) Gradient() (in module tvm.relax.transform) greater() (in module tvm.relax.op) greater_equal() (in module tvm.relax.op) grid_sample() (in module tvm.relax.op.image) (in module tvm.topi.image) grid_sample_python() (in module tvm.topi.testing) group() (tvm.testing.Path method) (tvm.tirx.TileLayout method) group_conv1d_ncw() (in module tvm.topi.nn) group_conv1d_ncw_python() (in module tvm.topi.testing) group_conv1d_nwc() (in module tvm.topi.nn) group_conv1d_transpose_ncw() (in module tvm.topi.nn) group_conv1d_transpose_ncw_python() (in module tvm.topi.testing) 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) group_norm_python() (in module tvm.topi.testing) GroupNormAttrs (class in tvm.relax.op.op_attrs) groups (tvm.relax.op.op_attrs.Conv1DAttrs property) (tvm.relax.op.op_attrs.Conv1DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv2DAttrs property) (tvm.relax.op.op_attrs.Conv2DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv3DAttrs property) (tvm.relax.op.op_attrs.Conv3DTransposeAttrs property) GT (class in tvm.tirx)

H

hamming_window() (in module tvm.relax.op) handle_add_byte_offset() (in module tvm.tirx) hardlink_to() (tvm.testing.Path method) has_attr() (tvm.ir.Op method) has_cutlass() (in module tvm.contrib.cutlass) has_reshape_pattern() (in module tvm.relax.analysis) has_workload() (tvm.s_tir.meta_schedule.Database method) have_cudagraph() (in module tvm.support.nvcc) have_fp16() (in module tvm.support.nvcc) have_int8() (in module tvm.support.nvcc) have_matrixcore() (in module tvm.support.rocm) have_tensorcore() (in module tvm.support.nvcc)

height (tvm.topi.nn.Workload attribute) hexagon() (tvm.rpc.RPCSession method) hexagon_clang_plus() (in module tvm.contrib.hexagon) hint_on_device() (in module tvm.relax.op) HintOnDeviceAttrs (class in tvm.relax.op.op_attrs) hoisted_conditionals (tvm.tirx.transform.HoistExpressionConfig property) hoisted_let_bindings (tvm.tirx.transform.HoistExpressionConfig property) HoistedConditionals (class in tvm.tirx.transform) HoistedLetBindings (class in tvm.tirx.transform) HoistExpressionConfig (class in tvm.tirx.transform) HoistIfThenElseConfig (class in tvm.tirx.transform) home() (tvm.testing.Path class method) hypot() (in module tvm.tirx)

I

Id (class in tvm.relax) id_index (tvm.relax.op.op_attrs.GetValidCountsAttrs property) (tvm.relax.op.op_attrs.NonMaximumSuppressionAttrs property) identity_after() (in module tvm.testing) If (class in tvm.relax) if_then_else() (in module tvm.te) (in module tvm.tirx) (in module tvm.topi.nn) IfElseExpr (tvm.tirx.transform.HoistedConditionals attribute) IfElseStmt (tvm.tirx.transform.HoistedConditionals attribute) IfThenElse (class in tvm.tirx) ignore_index (tvm.relax.op.op_attrs.NLLLossAttrs property) ignore_loop_partition() (in module tvm.tirx) ignore_profiler_call (tvm.tirx.transform.RemoveNoOpConfig property) 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) in_group (tvm.relax.op.op_attrs.AllGatherAttrs property) (tvm.relax.op.op_attrs.AllReduceAttrs property) index (tvm.relax.op.op_attrs.HintOnDeviceAttrs property) index() (tvm.topi.nn.Sequence method) index_map (tvm.relax.op.op_attrs.LayoutTransformAttrs property) index_put() (in module tvm.relax.op) index_tensor() (in module tvm.relax.op) indexdiv() (in module tvm.te) (in module tvm.tirx) indexing (tvm.relax.op.op_attrs.MeshgridAttrs property) IndexMap (class in tvm.tirx) indexmod() (in module tvm.te) (in module tvm.tirx) IndexPutAttrs (class in tvm.relax.op.op_attrs) indices_or_sections (tvm.relax.op.op_attrs.SplitAttrs property) infinity() (in module tvm.tirx) info (tvm.ir.transform.Pass property) init_ccl() (tvm.runtime.disco.Session method) InitAttrs (class in tvm.relax.op.op_attrs) InjectTextureAlloc() (in module tvm.s_tir.backend.adreno.transform) inline_mark() (tvm.s_tir.schedule.StmtSRef static method) InlinePrivateFunctions() (in module tvm.relax.transform) (in module tvm.tirx.transform) input_axis_separators (tvm.relax.op.op_attrs.LayoutTransformAttrs property) inputs (tvm.s_tir.schedule.Instruction attribute) install_request_hook() (in module tvm.testing) instance_norm() (in module tvm.relax.op.nn) (in module tvm.topi.nn) instance_norm_python() (in module tvm.topi.testing) InstanceNormAttrs (class in tvm.relax.op.op_attrs) Instruction (class in tvm.s_tir.schedule) InstructionKind (class in tvm.s_tir.schedule) insts (tvm.s_tir.schedule.Trace attribute) int_set() (tvm.arith.Analyzer method) InternalError IntervalSet (class in tvm.arith) IntImm (class in tvm.tirx) IntSet (class in tvm.arith) invalid_to_bottom (tvm.relax.op.op_attrs.NonMaximumSuppressionAttrs property) inverse() (tvm.tirx.IndexMap method) inverse_affine_iter_map() (in module tvm.arith) invoke_closure() (in module tvm.relax.op) (tvm.relax.VirtualMachine method) (tvm.runtime.vm.VirtualMachine method) invoke_pure_closure() (in module tvm.relax.op) invoke_stateful() (tvm.relax.VirtualMachine method) (tvm.runtime.vm.VirtualMachine method) iou_threshold (tvm.relax.op.op_attrs.NonMaximumSuppressionAttrs property)

IPCAllReduceRewrite (class in tvm.relax.transform) ir_module() (in module tvm.script) (in module tvm.script.ir_builder.ir) (in module tvm.script.parser) ir_transform() (in module tvm.tirx.stmt_functor) IRBuilder (class in tvm.script.ir_builder) IRModule (class in tvm.ir) IRModuleFrame (class in tvm.script.ir_builder.ir) is_alive() (tvm.support.popen_pool.PopenWorker method) is_ampere_or_newer() (in module tvm.testing) is_available() (in module tvm.contrib.nnpack) is_base_of() (tvm.ir.Type method) is_block_device() (tvm.testing.Path method) is_char_device() (tvm.testing.Path method) is_convolution() (tvm.s_tir.dlight.SBlockInfo method) is_data_pad() (tvm.s_tir.dlight.SBlockInfo method) is_dir() (tvm.testing.Path method) is_direct_sum_left() (tvm.tirx.Layout method) is_direct_sum_right() (tvm.tirx.Layout method) is_elementwise() (tvm.s_tir.dlight.SBlockInfo method) is_equivalent_to() (tvm.tirx.IndexMap method) is_everything() (tvm.arith.IntSet method) is_fifo() (tvm.testing.Path method) is_file() (tvm.testing.Path method) is_gemm() (tvm.s_tir.dlight.SBlockInfo method) is_gemv() (tvm.s_tir.dlight.SBlockInfo method) is_in_scope() (tvm.script.ir_builder.IRBuilder static method) is_injective() (tvm.s_tir.dlight.SBlockInfo method) is_layout_transform() (tvm.s_tir.dlight.SBlockInfo method) is_mount() (tvm.testing.Path method) is_nothing() (tvm.arith.IntSet method) is_pool() (tvm.s_tir.dlight.SBlockInfo method) is_pure (tvm.s_tir.schedule.InstructionKind property) is_pure_function() (in module tvm.s_tir.analysis) is_reduction() (tvm.s_tir.dlight.SBlockInfo method) is_scalar() (tvm.tirx.Buffer method) is_socket() (tvm.testing.Path method) is_source_path() (in module tvm.support.utils) is_swizzle() (tvm.tirx.Layout method) is_symlink() (tvm.testing.Path method) is_target_available() (tvm.s_tir.dlight.ScheduleRule method) is_tile_inner() (tvm.tirx.Layout method) is_tile_outer() (tvm.tirx.Layout method) is_trainium() (tvm.tirx.Layout method) is_trivial() (tvm.tirx.Layout method) (tvm.tirx.TileLayout method) is_z3_enabled (tvm.arith.Analyzer property) isfinite() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) isinf() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) isnan() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) isnullptr() (in module tvm.tirx) items() (tvm.ir.DictAttrs method) (tvm.ir.Map method) iter_map_simplify() (in module tvm.arith) iterdir() (tvm.testing.Path method) IterInfo (class in tvm.s_tir.dlight) IterMapExpr (class in tvm.arith) IterMark (class in tvm.arith) IterSplitExpr (class in tvm.arith) IterSumExpr (class in tvm.arith) IterVar (class in tvm.tirx)

J

join() (tvm.support.popen_pool.PopenWorker method)

join_running_task() (tvm.s_tir.meta_schedule.TaskScheduler method)

K

k (tvm.relax.op.op_attrs.TopKAttrs property) (tvm.relax.op.op_attrs.TriluAttrs property) keep_background (tvm.relax.op.op_attrs.MultiboxTransformLocAttrs property) keepdims (tvm.relax.op.op_attrs.ArgmaxArgminAttrs property) (tvm.relax.op.op_attrs.StatisticalAttrs property) kernel_h (tvm.topi.nn.Workload attribute) kernel_layout (tvm.relax.op.op_attrs.Conv1DAttrs property) (tvm.relax.op.op_attrs.Conv1DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv2DAttrs property) (tvm.relax.op.op_attrs.Conv2DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv3DAttrs property) (tvm.relax.op.op_attrs.Conv3DTransposeAttrs property)

kernel_w (tvm.topi.nn.Workload attribute) keys() (tvm.ir.DictAttrs method) (tvm.ir.Map method) kill() (tvm.support.popen_pool.PopenWorker method) kill_child_processes() (in module tvm.support.popen_pool) kill_object() (in module tvm.relax.op.vm) kill_storage() (in module tvm.relax.op.memory) kill_tensor() (in module tvm.relax.op.memory) KillAfterLastUse() (in module tvm.relax.transform) kind (tvm.s_tir.schedule.Instruction attribute)

L

l2_normalize_python() (in module tvm.topi.testing) LambdaLift() (in module tvm.relax.transform) largest (tvm.relax.op.op_attrs.TopKAttrs property) layer_norm() (in module tvm.relax.op.nn) (in module tvm.topi.nn) layer_norm_python() (in module tvm.topi.testing) LayerNormAttrs (class in tvm.relax.op.op_attrs) Layout (class in tvm.tirx) layout (tvm.relax.op.op_attrs.AdaptivePool1DAttrs property) (tvm.relax.op.op_attrs.AdaptivePool2DAttrs property) (tvm.relax.op.op_attrs.AdaptivePool3DAttrs property) (tvm.relax.op.op_attrs.Pool1DAttrs property) (tvm.relax.op.op_attrs.Pool2DAttrs property) (tvm.relax.op.op_attrs.Pool3DAttrs property) (tvm.relax.op.op_attrs.Resize2DAttrs property) (tvm.relax.op.op_attrs.ROIAlignAttrs property) (tvm.relax.op.op_attrs.ROIPoolAttrs property) 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) lchmod() (tvm.testing.Path method) ldexp() (in module tvm.tirx) LE (class in tvm.tirx) leaky_relu() (in module tvm.topi.nn) leakyrelu() (in module tvm.relax.op.nn) LeakyReluAttrs (class in tvm.relax.op.op_attrs) left_shift() (in module tvm.relax.op) LegalizeOps() (in module tvm.relax.transform) less() (in module tvm.relax.op) less_equal() (in module tvm.relax.op) Let (class in tvm.tirx) LetExpr (tvm.tirx.transform.HoistedLetBindings attribute) LetStmt (in module tvm.tirx) libinfo() (in module tvm.support) LiftTransformParams() (in module tvm.relax.transform) likely() (in module tvm.tirx) linear() (in module tvm.relax.op) link_shared_macos() (in module tvm.contrib.hexagon) link_to() (tvm.testing.Path method) list_configs() (tvm.ir.transform.PassContext static method) list_functions() (tvm.relax.BasePyModule method) list_kinds() (tvm.target.Target static method) list_op_names() (tvm.ir.Op static method) list_tags() (in module tvm.target) (in module tvm.target.tag) listdir() (tvm.rpc.RPCSession method) (tvm.support.utils.TempDirectory method)

load() (tvm.s_tir.meta_schedule.CostModel method) load_json() (in module tvm.ir) load_module() (tvm.rpc.RPCSession method) load_params() (tvm.relax.training.Trainer method) load_states() (tvm.relax.training.Trainer method) load_tensor_cache() (in module tvm.contrib.tvmjs) load_vm_module() (tvm.runtime.disco.Session method) local() (tvm.tirx.Buffer method) local_run() (in module tvm.testing) LocalSession (class in tvm.rpc) log() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) log10() (in module tvm.te) (in module tvm.tirx) log1p() (in module tvm.tirx) log2() (in module tvm.te) (in module tvm.tirx) log_add_exp() (in module tvm.relax.op) log_softmax() (in module tvm.contrib.cudnn) (in module tvm.relax.op.nn) (in module tvm.topi.nn) log_softmax_python() (in module tvm.topi.testing) logaddexp() (in module tvm.te) (in module tvm.tirx) 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_name() (in module tvm.script.ir_builder.ir) lookup_param() (in module tvm.tirx) lookup_vdevice() (in module tvm.script.ir_builder.ir) loop_partition() (tvm.s_tir.schedule.Schedule method) LoopRV (class in tvm.s_tir.schedule) LowerAllocTensor() (in module tvm.relax.transform) LowerGPUIPCAllocStorage (class in tvm.relax.transform) LowerIntrin() (in module tvm.tirx.transform) LowerRuntimeBuiltin() (in module tvm.relax.transform) LowerTIRx() (in module tvm.tirx.transform) LowerTIRxOpaque() (in module tvm.tirx.transform) LowerTVMBuiltin() (in module tvm.tirx.transform) LowerWarpMemory() (in module tvm.tirx.transform) lrn() (in module tvm.topi.nn) lrn_python() (in module tvm.topi.testing) lstat() (tvm.testing.Path method) lstm() (in module tvm.topi.nn) lstm_python() (in module tvm.topi.testing) LT (class in tvm.tirx)

M

make_closure() (in module tvm.relax.op) make_node() (in module tvm.ir) MakePackedAPI() (in module tvm.tirx.transform) Map (class in tvm.ir) map_indices() (tvm.tirx.IndexMap method) map_shape() (tvm.tirx.IndexMap method) map_tensor() (tvm.tirx.IndexMap method) map_with_error_catching() (tvm.support.popen_pool.PopenPoolExecutor method) MapResult (class in tvm.support.popen_pool) masked_fill() (in module tvm.relax.op) match_cast() (tvm.relax.block_builder.BlockBuilder method) MatchBufferRegion (class in tvm.tirx) MatchCast (class in tvm.relax) matmul() (in module tvm.contrib.cblas) (in module tvm.contrib.cublas) (in module tvm.contrib.cublaslt) (in module tvm.contrib.dnnl) (in module tvm.contrib.hipblas) (in module tvm.contrib.mkl) (in module tvm.relax.op) (in module tvm.topi.nn) matmul_u8s8s32() (in module tvm.contrib.mkl) MatmulAttrs (class in tvm.relax.op.op_attrs) matrix_set_diag() (in module tvm.topi.testing) Max (class in tvm.tirx) max() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) max_output_size (tvm.relax.op.op_attrs.NonMaximumSuppressionAttrs 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_simplification_steps (tvm.tirx.transform.RemoveNoOpConfig property) max_value() (in module tvm.te) (in module tvm.tirx) maximum() (in module tvm.relax.op) mean() (in module tvm.relax.op) MeasureCallback (class in tvm.s_tir.meta_schedule) MeasureCandidate (class in tvm.s_tir.meta_schedule) median() (in module tvm.relax.op) memoize() (in module tvm.contrib.pickle_memoize) memory_scope (tvm.relax.op.op_attrs.HintOnDeviceAttrs property) (tvm.target.VirtualDevice property) merge() (tvm.s_tir.schedule.Schedule method) MergeCompositeFunctions() (in module tvm.relax.transform) meshgrid() (in module tvm.relax.op) MeshgridAttrs (class in tvm.relax.op.op_attrs) metal() (tvm.rpc.RPCSession method) MetaScheduleApplyDatabase() (in module tvm.relax.transform) MetaScheduleTuneIRMod() (in module tvm.relax.transform) MetaScheduleTuneTIR() (in module tvm.relax.transform) method (tvm.relax.op.op_attrs.Resize2DAttrs property) Min (class in tvm.tirx) min() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) min_value() (in module tvm.te) (in module tvm.tirx) minimum() (in module tvm.relax.op) mirror_pad() (in module tvm.topi.nn) mkdir() (tvm.testing.Path method) mma_fill() (in module tvm.backend.cuda.op) mma_fill_legacy() (in module tvm.backend.cuda.op) mma_store() (in module tvm.backend.cuda.op) mma_store_legacy() (in module tvm.backend.cuda.op) Mod (class in tvm.tirx) mod (tvm.s_tir.schedule.Schedule property) mod() (in module tvm.relax.op) mode (tvm.relax.op.op_attrs.ROIAlignAttrs property) (tvm.relax.op.op_attrs.TakeAttrs property) modular_set() (tvm.arith.Analyzer method) ModularSet (class in tvm.arith) module tvm.arith tvm.backend.cuda tvm.backend.cuda.lang tvm.backend.cuda.op tvm.backend.cuda.operator tvm.backend.cuda.script tvm.backend.cuda.target_tags tvm.contrib tvm.contrib.cblas tvm.contrib.coreml_runtime tvm.contrib.cublas tvm.contrib.cublaslt tvm.contrib.cudnn tvm.contrib.cutlass tvm.contrib.dlpack tvm.contrib.dnnl tvm.contrib.download tvm.contrib.hexagon tvm.contrib.hipblas tvm.contrib.mkl tvm.contrib.nnpack tvm.contrib.pickle_memoize tvm.contrib.random tvm.contrib.thrust tvm.contrib.tvmjs tvm.driver tvm.error tvm.exec tvm.ir tvm.ir.instrument tvm.ir.transform tvm.relax tvm.relax.analysis tvm.relax.backend tvm.relax.block_builder tvm.relax.distributed tvm.relax.dpl tvm.relax.frontend tvm.relax.frontend.onnx tvm.relax.frontend.stablehlo tvm.relax.frontend.tflite 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.op.vision tvm.relax.op.vm tvm.relax.training tvm.relax.transform tvm.relax.transform.legalize_ops tvm.rpc tvm.runtime tvm.runtime.disco tvm.runtime.vm tvm.s_tir.analysis tvm.s_tir.backend tvm.s_tir.backend.adreno tvm.s_tir.backend.adreno.pipeline tvm.s_tir.backend.adreno.transform tvm.s_tir.dlight tvm.s_tir.meta_schedule tvm.s_tir.schedule tvm.s_tir.tensor_intrin tvm.script tvm.script.ir_builder tvm.script.ir_builder.ir tvm.script.ir_builder.relax tvm.script.ir_builder.relax.distributed tvm.script.ir_builder.tirx tvm.script.parser tvm.script.parser.ir tvm.script.parser.relax tvm.script.parser.tirx tvm.script.printer tvm.script.relax tvm.script.tirx tvm.support tvm.support.cc tvm.support.clang tvm.support.emcc tvm.support.ndk tvm.support.nvcc tvm.support.popen_pool tvm.support.rocm tvm.support.tar tvm.support.utils tvm.support.xcode tvm.target tvm.target.tag tvm.te tvm.testing tvm.tirx tvm.tirx.analysis.analysis tvm.tirx.stmt_functor tvm.tirx.transform tvm.topi.cpp tvm.topi.image tvm.topi.nn tvm.topi.testing

module (tvm.contrib.coreml_runtime.CoreMLModule attribute) module_attrs() (in module tvm.script.ir_builder.ir) module_get_attr() (in module tvm.script.ir_builder.ir) module_global_infos() (in module tvm.script.ir_builder.ir) module_pass() (in module tvm.ir.transform) module_set_attr() (in module tvm.script.ir_builder.ir) ModulePass (class in tvm.ir.transform) momentum (tvm.relax.op.op_attrs.BatchNormAttrs property) Mul (class in tvm.tirx) multibox_transform_loc() (in module tvm.relax.op) (in module tvm.relax.op.vision) MultiboxTransformLocAttrs (class in tvm.relax.op.op_attrs) multinomial_from_uniform() (in module tvm.relax.op) MultinomialFromUniformAttrs (class in tvm.relax.op.op_attrs) multiply() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) mutate_irmodule() (tvm.relax.DataflowBlockRewrite method) mutated_dfb() (tvm.relax.DataflowBlockRewrite method) mutated_root_fn() (tvm.relax.DataflowBlockRewrite method) Mutator (class in tvm.s_tir.meta_schedule)

N

name (tvm.s_tir.schedule.InstructionKind attribute) (tvm.tirx.ExecScope property) name() (tvm.script.ir_builder.IRBuilder static method) name_hint (tvm.relax.Var property) name_many() (tvm.script.ir_builder.IRBuilder static method) name_to_binding() (in module tvm.relax.analysis) namedtuple() (in module tvm.topi.nn) NarrowDataType() (in module tvm.tirx.transform) nchw_pack_layout() (in module tvm.topi.image) nchw_xc_layout() (in module tvm.topi.image) ndim (tvm.te.Tensor property) NE (class in tvm.tirx) nearbyint() (in module tvm.te) (in module tvm.tirx) negative() (in module tvm.relax.op) Never (tvm.tirx.transform.HoistedConditionals attribute) (tvm.tirx.transform.HoistedLetBindings attribute) next_task_id() (tvm.s_tir.meta_schedule.TaskScheduler method) nextafter() (in module tvm.tirx) nll_loss() (in module tvm.relax.op.nn) (in module tvm.topi.nn) (in module tvm.topi.testing) nll_loss_backward() (in module tvm.relax.op.grad) NLLLossAttrs (class in tvm.relax.op.op_attrs) no_grad() (in module tvm.relax.op.grad) Node (class in tvm.ir) non_max_suppression() (in module tvm.relax.op) (in module tvm.relax.op.vision) non_max_suppression_python() (in module tvm.topi.testing) non_surjective_inverse() (tvm.tirx.IndexMap method) NonMaximumSuppressionAttrs (class in tvm.relax.op.op_attrs) 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.support.tar) normalize_iter_map_to_expr() (in module tvm.arith) normalize_prim_func() (in module tvm.s_tir.dlight) normalize_to_iter_sum() (in module tvm.arith) NormalizeGlobalVar() (in module tvm.relax.transform) Not (class in tvm.tirx) not_equal() (in module tvm.relax.op) notify_runner_results() (tvm.s_tir.meta_schedule.SearchStrategy method) (tvm.s_tir.meta_schedule.TuneContext method) np_dtype_from_str() (in module tvm.testing) null_value() (in module tvm.relax.op) num_groups (tvm.relax.op.op_attrs.GroupNormAttrs property) num_workers (tvm.relax.op.op_attrs.AllGatherAttrs property) (tvm.relax.op.op_attrs.ScatterCollectiveAttrs property) (tvm.runtime.disco.Session property) nvshmem_barrier_all() (in module tvm.backend.cuda.op) nvshmem_fence() (in module tvm.backend.cuda.op) nvshmem_getmem_nbi() (in module tvm.backend.cuda.op) nvshmem_getmem_nbi_block() (in module tvm.backend.cuda.op) nvshmem_getmem_nbi_warp() (in module tvm.backend.cuda.op) nvshmem_my_pe() (in module tvm.backend.cuda.op) nvshmem_n_pes() (in module tvm.backend.cuda.op) nvshmem_putmem_nbi() (in module tvm.backend.cuda.op) nvshmem_putmem_nbi_block() (in module tvm.backend.cuda.op) nvshmem_putmem_nbi_warp() (in module tvm.backend.cuda.op) nvshmem_putmem_signal_nbi() (in module tvm.backend.cuda.op) nvshmem_putmem_signal_nbi_block() (in module tvm.backend.cuda.op) nvshmem_putmem_signal_nbi_warp() (in module tvm.backend.cuda.op) nvshmem_quiet() (in module tvm.backend.cuda.op) nvshmem_signal_op() (in module tvm.backend.cuda.op) nvshmem_wait_until() (in module tvm.backend.cuda.op) NVSHMEMNamespace (class in tvm.backend.cuda.script)

O

ObjectType (class in tvm.relax) offset_of() (tvm.tirx.Buffer method) one_hot() (in module tvm.relax.op) (in module tvm.topi.testing) OneHotAttrs (class in tvm.relax.op.op_attrs) ones() (in module tvm.relax.op) ones_like() (in module tvm.relax.op) OOBChecker() (in module tvm.s_tir.analysis) Op (class in tvm.ir) op_type (tvm.relax.op.op_attrs.AllReduceAttrs property) OPAQUE (tvm.s_tir.schedule.DepKind attribute) OpAttributeInvalid OpAttributeRequired OpAttributeUnImplemented open() (tvm.testing.Path method) OpError OpNotImplemented OptimizeLayoutTransform (class in tvm.relax.transform) options (tvm.target.TargetKind property) options_from_name() (tvm.target.TargetKind static method) Or (class in tvm.tirx) out_dtype (tvm.relax.op.op_attrs.Conv1DAttrs property) (tvm.relax.op.op_attrs.Conv1DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv2DAttrs property) (tvm.relax.op.op_attrs.Conv2DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv3DAttrs property) (tvm.relax.op.op_attrs.Conv3DTransposeAttrs property) (tvm.relax.op.op_attrs.MatmulAttrs property) (tvm.relax.op.op_attrs.QuantizeAttrs property) (tvm.relax.op.op_attrs.Resize2DAttrs property) (tvm.topi.nn.Workload attribute)

out_filter (tvm.topi.nn.Workload attribute) out_layout (tvm.relax.op.op_attrs.AdaptivePool1DAttrs property) (tvm.relax.op.op_attrs.AdaptivePool2DAttrs property) (tvm.relax.op.op_attrs.AdaptivePool3DAttrs property) (tvm.relax.op.op_attrs.Conv1DAttrs property) (tvm.relax.op.op_attrs.Conv1DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv2DAttrs property) (tvm.relax.op.op_attrs.Conv2DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv3DAttrs property) (tvm.relax.op.op_attrs.Conv3DTransposeAttrs property) (tvm.relax.op.op_attrs.Pool1DAttrs property) (tvm.relax.op.op_attrs.Pool2DAttrs property) (tvm.relax.op.op_attrs.Pool3DAttrs property) outer() (in module tvm.relax.op) output_format (tvm.relax.op.op_attrs.AllClassNonMaximumSuppressionAttrs property) output_padding (tvm.relax.op.op_attrs.Conv1DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv2DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv3DTransposeAttrs property) output_size (tvm.relax.op.op_attrs.AdaptivePool1DAttrs property) (tvm.relax.op.op_attrs.AdaptivePool2DAttrs property) (tvm.relax.op.op_attrs.AdaptivePool3DAttrs property) outputs (tvm.s_tir.schedule.Instruction attribute) override_instruments() (tvm.ir.transform.PassContext method) owner() (tvm.testing.Path method)

P

pack() (tvm.tirx.Layout method) pack_imports() (in module tvm.contrib.hexagon) 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.s_tir.schedule.Schedule method) pad_mode (tvm.relax.op.op_attrs.PadAttrs property) pad_value (tvm.relax.op.op_attrs.LayoutTransformAttrs property) (tvm.relax.op.op_attrs.PadAttrs property) pad_width (tvm.relax.op.op_attrs.PadAttrs property) PadAttrs (class in tvm.relax.op.op_attrs) padb (tvm.topi.nn.Workload attribute) padding (tvm.relax.op.op_attrs.Conv1DAttrs property) (tvm.relax.op.op_attrs.Conv1DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv2DAttrs property) (tvm.relax.op.op_attrs.Conv2DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv3DAttrs property) (tvm.relax.op.op_attrs.Conv3DTransposeAttrs property) (tvm.relax.op.op_attrs.Pool1DAttrs property) (tvm.relax.op.op_attrs.Pool2DAttrs property) (tvm.relax.op.op_attrs.Pool3DAttrs property) padl (tvm.topi.nn.Workload attribute) padr (tvm.topi.nn.Workload attribute) padt (tvm.topi.nn.Workload attribute) parallel() (tvm.s_tir.schedule.Schedule method) parameter() (in module tvm.testing) parent (tvm.s_tir.schedule.StmtSRef property) parse() (in module tvm.script.parser) parse_compute_version() (in module tvm.support.nvcc) (in module tvm.support.rocm) Pass (class in tvm.ir.transform) pass_instrument() (in module tvm.ir.instrument) PassContext (class in tvm.ir.transform) PassInfo (class in tvm.ir.transform) PassInstrument (class in tvm.ir.instrument) PassPrintingInstrument (class in tvm.ir.instrument) PassTimingInstrument (class in tvm.ir.instrument) Path (class in tvm.testing) PatternCheckContext (class in tvm.relax.transform) pending_nbytes (tvm.contrib.tvmjs.TensorCacheShardingManager property) permute() (tvm.tirx.Buffer method) permute_by_groups() (tvm.tirx.TileLayout method) permute_dims() (in module tvm.relax.op) (tvm.tirx.TileLayout method) PermuteDimsAttrs (class in tvm.relax.op.op_attrs) pixel_shuffle() (in module tvm.relax.op.nn) PixelShuffleAttrs (class in tvm.relax.op.op_attrs) placeholder() (in module tvm.te) PlaceholderOp (class in tvm.te) Placement (class in tvm.relax.distributed) PlacementSpec (class in tvm.relax.distributed) PointerType (class in tvm.ir) PointerValueTypeRewrite() (in module tvm.tirx.transform) pool1d() (in module tvm.topi.nn) Pool1DAttrs (class in tvm.relax.op.op_attrs) pool2d() (in module tvm.topi.nn) Pool2DAttrs (class in tvm.relax.op.op_attrs) pool3d() (in module tvm.topi.nn) Pool3DAttrs (class in tvm.relax.op.op_attrs) pool_grad() (in module tvm.topi.nn) pool_grad_nchw() (in module tvm.topi.testing) pool_size (tvm.relax.op.op_attrs.Pool1DAttrs property) (tvm.relax.op.op_attrs.Pool2DAttrs property) (tvm.relax.op.op_attrs.Pool3DAttrs property) pooled_size (tvm.relax.op.op_attrs.ROIAlignAttrs property) (tvm.relax.op.op_attrs.ROIPoolAttrs property) poolnd_python() (in module tvm.topi.testing) pop() (tvm.s_tir.schedule.Trace method) popcount() (in module tvm.te) (in module tvm.tirx) PopenPoolExecutor (class in tvm.support.popen_pool) PopenSession (class in tvm.rpc) PopenWorker (class in tvm.support.popen_pool) post_order_visit() (in module tvm.relax.analysis) (in module tvm.tirx.stmt_functor) post_tuning() (tvm.s_tir.meta_schedule.SearchStrategy method) (tvm.s_tir.meta_schedule.TuneContext method) Postproc (class in tvm.s_tir.meta_schedule) pow() (in module tvm.tirx) power() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) pre_order_visit() (in module tvm.tirx.stmt_functor) pre_tuning() (tvm.s_tir.meta_schedule.SearchStrategy method) (tvm.s_tir.meta_schedule.TuneContext method) Predicate (class in tvm.tirx) predict() (tvm.relax.training.Trainer method) (tvm.s_tir.meta_schedule.CostModel method) prelu() (in module tvm.relax.op.nn) (in module tvm.topi.nn) PReluAttrs (class in tvm.relax.op.op_attrs) PresburgerSet (class in tvm.arith) prim_func_pass() (in module tvm.tirx.transform) PrimExpr (class in tvm.ir) PrimFunc (class in tvm.tirx) PrimFuncPass (class in tvm.tirx.transform) PrimType (class in tvm.ir) PrimValue (class in tvm.relax) print() (in module tvm.relax.op) print_tuning_statistics() (tvm.s_tir.meta_schedule.TaskScheduler method) PrintAfterAll (class in tvm.ir.instrument) PrintBeforeAll (class in tvm.ir.instrument) PrintIR() (in module tvm.ir.transform)

ProcessSession (class in tvm.runtime.disco) prod() (in module tvm.relax.op) ProducerLoad (class in tvm.tirx) Profiler (class in tvm.s_tir.meta_schedule) ProofStrength (class in tvm.arith) ptr_byte_offset() (in module tvm.tirx) ptr_to() (tvm.tirx.Buffer method) ptx_add_f32() (in module tvm.backend.cuda.op) ptx_add_f32x2() (in module tvm.backend.cuda.op) ptx_add_f64() (in module tvm.backend.cuda.op) ptx_any_sync() (in module tvm.backend.cuda.op) ptx_bar_arrive() (in module tvm.backend.cuda.op) ptx_bar_sync() (in module tvm.backend.cuda.op) ptx_barrier_cluster_arrive() (in module tvm.backend.cuda.op) ptx_barrier_cluster_wait() (in module tvm.backend.cuda.op) ptx_clc_query_cancel() (in module tvm.backend.cuda.op) ptx_clc_try_cancel() (in module tvm.backend.cuda.op) ptx_cp_async() (in module tvm.backend.cuda.op) ptx_cp_async_bulk() (in module tvm.backend.cuda.op) ptx_cp_async_bulk_commit_group() (in module tvm.backend.cuda.op) ptx_cp_async_bulk_shared_to_cluster() (in module tvm.backend.cuda.op) ptx_cp_async_bulk_tensor_global_to_cluster() (in module tvm.backend.cuda.op) ptx_cp_async_bulk_tensor_global_to_cluster_prefetch() (in module tvm.backend.cuda.op) ptx_cp_async_bulk_tensor_shared_to_global() (in module tvm.backend.cuda.op) ptx_cp_async_bulk_tensor_shared_to_global_reduce() (in module tvm.backend.cuda.op) ptx_cp_async_bulk_tensor_tile_gather4_global_to_cluster() (in module tvm.backend.cuda.op) ptx_cp_async_bulk_wait_group() (in module tvm.backend.cuda.op) ptx_cp_async_commit_group() (in module tvm.backend.cuda.op) ptx_cp_async_legacy() (in module tvm.backend.cuda.op) ptx_cp_async_mbarrier_arrive() (in module tvm.backend.cuda.op) ptx_cp_async_wait_group() (in module tvm.backend.cuda.op) ptx_elect_sync() (in module tvm.backend.cuda.op) ptx_exp2() (in module tvm.backend.cuda.op) ptx_fence() (in module tvm.backend.cuda.op) ptx_fence_mbarrier_init() (in module tvm.backend.cuda.op) ptx_fence_proxy_async() (in module tvm.backend.cuda.op) ptx_fetch_register() (in module tvm.backend.cuda.op) ptx_fma_f32() (in module tvm.backend.cuda.op) ptx_fma_f32x2() (in module tvm.backend.cuda.op) ptx_fma_f64() (in module tvm.backend.cuda.op) ptx_griddepcontrol_launch_dependents() (in module tvm.backend.cuda.op) ptx_griddepcontrol_wait() (in module tvm.backend.cuda.op) ptx_ld() (in module tvm.backend.cuda.op) ptx_ld_acquire() (in module tvm.backend.cuda.op) ptx_ld_global_acquire() (in module tvm.backend.cuda.op) ptx_ld_volatile() (in module tvm.backend.cuda.op) ptx_ldmatrix() (in module tvm.backend.cuda.op) ptx_ldmatrix_legacy() (in module tvm.backend.cuda.op) ptx_map_shared_rank() (in module tvm.backend.cuda.op) ptx_mapa() (in module tvm.backend.cuda.op) ptx_max_f32() (in module tvm.backend.cuda.op) ptx_mbarrier_arrive() (in module tvm.backend.cuda.op) ptx_mbarrier_arrive_cluster_count() (in module tvm.backend.cuda.op) ptx_mbarrier_arrive_expect_tx() (in module tvm.backend.cuda.op) ptx_mbarrier_init() (in module tvm.backend.cuda.op) ptx_mbarrier_try_wait() (in module tvm.backend.cuda.op) ptx_mbarrier_try_wait_acquire_cluster() (in module tvm.backend.cuda.op) ptx_mbarrier_try_wait_once() (in module tvm.backend.cuda.op) ptx_mma() (in module tvm.backend.cuda.op) ptx_mma_legacy() (in module tvm.backend.cuda.op) ptx_mma_sp() (in module tvm.backend.cuda.op) ptx_mma_sp_legacy() (in module tvm.backend.cuda.op) ptx_mul_f32() (in module tvm.backend.cuda.op) ptx_mul_f32x2() (in module tvm.backend.cuda.op) ptx_mul_f64() (in module tvm.backend.cuda.op) ptx_rcp() (in module tvm.backend.cuda.op) ptx_reduce3_max_f32() (in module tvm.backend.cuda.op) ptx_reduce3_min_f32() (in module tvm.backend.cuda.op) ptx_setmaxnreg() (in module tvm.backend.cuda.op) ptx_stmatrix() (in module tvm.backend.cuda.op) ptx_sub_f32() (in module tvm.backend.cuda.op) ptx_sub_f32x2() (in module tvm.backend.cuda.op) ptx_sub_f64() (in module tvm.backend.cuda.op) ptx_tcgen05_alloc() (in module tvm.backend.cuda.op) ptx_tcgen05_commit() (in module tvm.backend.cuda.op) ptx_tcgen05_cp() (in module tvm.backend.cuda.op) ptx_tcgen05_dealloc() (in module tvm.backend.cuda.op) ptx_tcgen05_encode_instr_descriptor() (in module tvm.backend.cuda.op) ptx_tcgen05_encode_instr_descriptor_block_scaled() (in module tvm.backend.cuda.op) ptx_tcgen05_encode_matrix_descriptor() (in module tvm.backend.cuda.op) ptx_tcgen05_fence_after_thread_sync() (in module tvm.backend.cuda.op) ptx_tcgen05_fence_before_thread_sync() (in module tvm.backend.cuda.op) ptx_tcgen05_ld() (in module tvm.backend.cuda.op) ptx_tcgen05_mma() (in module tvm.backend.cuda.op) ptx_tcgen05_mma_block_scale() (in module tvm.backend.cuda.op) ptx_tcgen05_mma_sp() (in module tvm.backend.cuda.op) ptx_tcgen05_mma_sp_block_scale() (in module tvm.backend.cuda.op) ptx_tcgen05_relinquish_alloc_permit() (in module tvm.backend.cuda.op) ptx_tcgen05_shift() (in module tvm.backend.cuda.op) ptx_tcgen05_st() (in module tvm.backend.cuda.op) ptx_tcgen05_wait_ld() (in module tvm.backend.cuda.op) ptx_tcgen05_wait_st() (in module tvm.backend.cuda.op) ptx_wgmma_commit_group() (in module tvm.backend.cuda.op) ptx_wgmma_encode_matrix_descriptor() (in module tvm.backend.cuda.op) ptx_wgmma_fence() (in module tvm.backend.cuda.op) ptx_wgmma_mma_async_rs() (in module tvm.backend.cuda.op) ptx_wgmma_mma_async_ss() (in module tvm.backend.cuda.op) ptx_wgmma_noop_barrier() (in module tvm.backend.cuda.op) ptx_wgmma_wait_group() (in module tvm.backend.cuda.op) PTXNamespace (class in tvm.backend.cuda.script) PyExprMutator (class in tvm.relax) PyExprVisitor (class in tvm.relax) PyStmtExprMutator (class in tvm.tirx) PyStmtExprVisitor (class in tvm.tirx)

Q

q_multiply_shift() (in module tvm.tirx) q_multiply_shift_per_axis() (in module tvm.tirx) quantize() (in module tvm.relax.op) QuantizeAttrs (class in tvm.relax.op.op_attrs)

query() (tvm.s_tir.meta_schedule.Database method) query_ir_module() (tvm.s_tir.meta_schedule.Database method) query_schedule() (tvm.s_tir.meta_schedule.Database method) query_tuning_record() (tvm.s_tir.meta_schedule.Database method)

R

r() (tvm.relax.ExecBuilder method) Ramp (class in tvm.tirx) randint() (in module tvm.contrib.random) Range (class in tvm.ir) rate (tvm.relax.op.op_attrs.DropoutAttrs property) RAW (tvm.s_tir.schedule.DepKind attribute) read_bytes() (tvm.testing.Path method) read_text() (tvm.testing.Path method) readlink() (tvm.testing.Path method) RealizeVDevice() (in module tvm.relax.transform) recv() (tvm.support.popen_pool.PopenWorker method) redistribute() (in module tvm.relax.op.distributed) redistribute_replica_to_shard() (in module tvm.relax.op.distributed) Reduce (class in tvm.tirx) reduce() (in module tvm.topi.nn) reduce_axis() (in module tvm.te) reduction (tvm.relax.op.op_attrs.NLLLossAttrs property) (tvm.relax.op.op_attrs.ScatterElementsAttrs property) (tvm.relax.op.op_attrs.ScatterNDAttrs property) reflect_pad() (in module tvm.topi.nn) register() (tvm.tirx.TensorIntrin static method) register_backend() (in module tvm.backend.cuda) register_dialect() (in module tvm.script) register_error() (in module tvm.error) register_global_func() (in module tvm.contrib.hexagon) register_gradient() (in module tvm.relax.op) register_intrin_lowering() (in module tvm.ir) register_linker() (in module tvm.contrib.hexagon) register_op_attr() (in module tvm.ir) register_pipeline() (in module tvm.relax) register_tag() (in module tvm.backend.cuda.target_tags) (in module tvm.target) (in module tvm.target.tag) register_tir_pipeline() (in module tvm.tirx) reindex() (tvm.s_tir.schedule.Schedule method) reindex_cache_read() (tvm.s_tir.schedule.Schedule method) reindex_cache_write() (tvm.s_tir.schedule.Schedule method) reinterpret() (in module tvm.tirx) relax_dynamo() (in module tvm.relax.frontend.torch) RelaxExpr (class in tvm.ir) release() (tvm.support.utils.FileLock method) relpath() (tvm.support.utils.TempDirectory method) relu() (in module tvm.relax.op.nn) (in module tvm.topi.nn) relu6() (in module tvm.relax.op.nn) remove() (tvm.rpc.RPCSession method) (tvm.support.utils.TempDirectory method) remove_all_unused() (in module tvm.relax.analysis) (tvm.relax.DataflowBlockRewrite method) remove_rv() (tvm.s_tir.schedule.Schedule method) remove_unused() (tvm.relax.DataflowBlockRewrite method) RemoveAssume() (in module tvm.tirx.transform) RemoveNoOp() (in module tvm.tirx.transform) RemoveNoOpConfig (class in tvm.tirx.transform) RemovePurityChecking() (in module tvm.relax.transform) RemoveRedundantReshape (class in tvm.relax.transform) RemoveUnusedOutputs() (in module tvm.relax.transform) RemoveUnusedParameters() (in module tvm.relax.transform) rename() (tvm.testing.Path method) render() (tvm.ir.instrument.PassTimingInstrument static method) renew_defs() (in module tvm.tirx.stmt_functor) reorder() (tvm.s_tir.schedule.Schedule method) reorder_block_iter_var() (tvm.s_tir.schedule.Schedule method) ReorderPermuteDimsAfterConcat() (in module tvm.relax.transform) ReorderTakeAfterMatmul() (in module tvm.relax.transform) reorg_python() (in module tvm.topi.testing) repeat() (in module tvm.relax.op) RepeatAttrs (class in tvm.relax.op.op_attrs)

repeats (tvm.relax.op.op_attrs.RepeatAttrs property) (tvm.relax.op.op_attrs.TileAttrs property) replace() (tvm.s_tir.schedule.ScheduleState method) (tvm.testing.Path method) (tvm.tirx.TilePrimitiveCall method) replace_all_uses() (tvm.relax.DataflowBlockRewrite method) replica() (tvm.relax.distributed.PlacementSpec static method) replicate_pad() (in module tvm.topi.nn) request() (tvm.rpc.TrackerSession method) request_and_run() (tvm.rpc.TrackerSession method) RequiredByConditional (tvm.tirx.transform.HoistedLetBindings attribute) reset_attr() (tvm.ir.Op method) reshape() (in module tvm.relax.op) resize1d() (in module tvm.topi.image) resize1d_python() (in module tvm.topi.testing) resize2d() (in module tvm.relax.op.image) (in module tvm.topi.image) resize2d_python() (in module tvm.topi.testing) Resize2DAttrs (class in tvm.relax.op.op_attrs) resize3d() (in module tvm.relax.op.image) (in module tvm.topi.image) resize3d_python() (in module tvm.topi.testing) resolve() (tvm.testing.Path method) ret() (in module tvm.tirx) ret_type (tvm.relax.op.op_attrs.TopKAttrs property) return_indices (tvm.relax.op.op_attrs.NonMaximumSuppressionAttrs property) reverse_compute_at() (tvm.s_tir.schedule.Schedule method) reverse_compute_inline() (tvm.s_tir.schedule.Schedule method) rewrite_simplify() (tvm.arith.Analyzer method) RewriteCUDAGraph() (in module tvm.relax.transform) RewriteDataflowReshape() (in module tvm.relax.transform) rfactor() (tvm.s_tir.schedule.Schedule method) rglob() (tvm.testing.Path method) right_shift() (in module tvm.relax.op) rmdir() (tvm.testing.Path method) rms_norm() (in module tvm.relax.op.nn) (in module tvm.topi.nn) rms_norm_python() (in module tvm.topi.testing) RMSNormAttrs (class in tvm.relax.op.op_attrs) rocm() (tvm.rpc.RPCSession method) rocm_link() (in module tvm.support.rocm) roi (tvm.relax.op.op_attrs.Resize2DAttrs property) roi_align() (in module tvm.relax.op) (in module tvm.relax.op.vision) roi_align_nchw_python() (in module tvm.topi.testing) roi_align_nhwc_python() (in module tvm.topi.testing) roi_pool() (in module tvm.relax.op) (in module tvm.relax.op.vision) roi_pool_nchw_python() (in module tvm.topi.testing) ROIAlignAttrs (class in tvm.relax.op.op_attrs) ROIPoolAttrs (class in tvm.relax.op.op_attrs) rolling_buffer() (tvm.s_tir.schedule.Schedule method) root_mark() (tvm.s_tir.schedule.StmtSRef static method) round() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) rounding_method (tvm.relax.op.op_attrs.Resize2DAttrs property) rpc_run() (in module tvm.testing) RPCError RPCSession (class in tvm.rpc) RPCSessionTimeoutError rsqrt() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) run() (tvm.s_tir.meta_schedule.Runner method) run_after_pass() (tvm.ir.instrument.PassInstrument method) run_before_pass() (tvm.ir.instrument.PassInstrument method) RunCodegen() (in module tvm.relax.transform) Runner (class in tvm.s_tir.meta_schedule)

S

same_as() (tvm.ir.GlobalInfo method) (tvm.ir.Type method) samefile() (tvm.testing.Path method) sample_categorical() (tvm.s_tir.schedule.Schedule method) sample_compute_location() (tvm.s_tir.schedule.Schedule method) sample_partitioned_tile() (tvm.s_tir.schedule.Schedule method) sample_perfect_tile() (tvm.s_tir.schedule.Schedule method) sample_ratio (tvm.relax.op.op_attrs.ROIAlignAttrs property) save() (tvm.s_tir.meta_schedule.CostModel method) save_function() (tvm.relax.VirtualMachine method) (tvm.runtime.vm.VirtualMachine method) save_json() (in module tvm.ir) SBlock (class in tvm.tirx) SBlockInfo (class in tvm.s_tir.dlight) SBlockRealize (class in tvm.tirx) SBlockRV (class in tvm.s_tir.schedule) SBlockScope (class in tvm.s_tir.schedule) scale (tvm.relax.op.op_attrs.AttentionAttrs property) (tvm.relax.op.op_attrs.BatchNormAttrs property) (tvm.relax.op.op_attrs.GroupNormAttrs property) (tvm.relax.op.op_attrs.InstanceNormAttrs property) (tvm.relax.op.op_attrs.LayerNormAttrs property) 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) 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) ScatterCollectiveAttrs (class in tvm.relax.op.op_attrs) ScatterElementsAttrs (class in tvm.relax.op.op_attrs) ScatterNDAttrs (class in tvm.relax.op.op_attrs) Schedule (class in tvm.s_tir.schedule) ScheduleDebugMask (class in tvm.s_tir.schedule) ScheduleError ScheduleRule (class in tvm.s_tir.dlight) (class in tvm.s_tir.meta_schedule) ScheduleState (class in tvm.s_tir.schedule) scope() (tvm.tirx.Buffer method) ScopeIdDef (class in tvm.tirx) ScopeIdDefStmt (class in tvm.tirx) score_index (tvm.relax.op.op_attrs.GetValidCountsAttrs property) (tvm.relax.op.op_attrs.NonMaximumSuppressionAttrs property) score_threshold (tvm.relax.op.op_attrs.GetValidCountsAttrs property) (tvm.relax.op.op_attrs.NonMaximumSuppressionAttrs property) script() (tvm.relax.BasePyModule method) script_namespace() (in module tvm.backend.cuda) script_namespaces() (in module tvm.backend.cuda) searchsorted_ref() (in module tvm.topi.testing) SearchStrategy (class in tvm.s_tir.meta_schedule) seed() (tvm.s_tir.schedule.Schedule method) Select (class in tvm.tirx) selu() (in module tvm.relax.op.nn) send() (tvm.support.popen_pool.PopenWorker method) SeqExpr (class in tvm.relax) SeqStmt (class in tvm.tirx) Sequence (class in tvm.topi.nn) sequence_mask() (in module tvm.topi.testing) Sequential (class in tvm.ir.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.s_tir.schedule.Schedule method) set_input() (tvm.relax.VirtualMachine method) (tvm.runtime.vm.VirtualMachine method) set_instrument() (tvm.relax.VirtualMachine method) (tvm.runtime.vm.VirtualMachine method) set_keep_for_debug() (tvm.support.utils.TempDirectory class method) set_maximum_rewrite_steps() (tvm.arith.Analyzer method) set_num_inputs() (tvm.ir.Op method) set_scope() (tvm.s_tir.schedule.Schedule method) set_support_level() (tvm.ir.Op method) set_var_remap() (tvm.relax.PyExprMutator method) set_z3_rlimit() (tvm.arith.Analyzer method) set_z3_timeout_ms() (tvm.arith.Analyzer method) SetupTrainer (class in tvm.relax.training) shape_of() (in module tvm.relax.op) shape_to_tensor() (in module tvm.relax.op) ShapeExpr (class in tvm.relax) ShapeType (class in tvm.relax) sharding() (tvm.relax.distributed.PlacementSpec static method) shift_left() (in module tvm.tirx) shift_right() (in module tvm.tirx) should_run() (tvm.ir.instrument.PassInstrument method) show() (tvm.relax.BasePyModule method) (tvm.s_tir.schedule.Schedule method) (tvm.s_tir.schedule.Trace method) Shuffle (class in tvm.tirx) shutdown() (tvm.runtime.disco.Session method) (tvm.support.popen_pool.PopenPoolExecutor method) sigmoid() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) sign() (in module tvm.relax.op) silu() (in module tvm.relax.op.nn) simplified() (tvm.s_tir.schedule.Trace method) simplify() (in module tvm.topi.image) (in module tvm.topi.nn) (tvm.arith.Analyzer method) 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.tirx) single_point() (tvm.arith.IntSet static method) sinh() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx)

size() (in module tvm.relax.op) (tvm.tirx.Layout method) size_var() (in module tvm.te) SizeVar (class in tvm.tirx) SkipAssert() (in module tvm.tirx.transform) slice() (tvm.tirx.Layout method) slice_axis_python() (in module tvm.topi.testing) slice_scatter() (in module tvm.relax.op) SliceScatterAttrs (class in tvm.relax.op.op_attrs) SocketSession (class in tvm.runtime.disco) soft_nms_sigma (tvm.relax.op.op_attrs.NonMaximumSuppressionAttrs property) softmax() (in module tvm.contrib.cudnn) (in module tvm.relax.op.nn) (in module tvm.topi.nn) softmax_common() (in module tvm.topi.nn) softmax_python() (in module tvm.topi.testing) SoftmaxAttrs (class in tvm.relax.op.op_attrs) softplus() (in module tvm.relax.op.nn) (in module tvm.topi.nn) SoftplusAttrs (class in tvm.relax.op.op_attrs) solve_linear_equations() (in module tvm.arith) solve_linear_inequalities() (in module tvm.arith) 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_batch_nd_python() (in module tvm.topi.testing) space_to_depth() (in module tvm.topi.nn) space_to_depth_python() (in module tvm.topi.testing) SpaceGenerator (class in tvm.s_tir.meta_schedule) Span (class in tvm.ir) span() (tvm.tirx.Layout method) spatial_scale (tvm.relax.op.op_attrs.ROIAlignAttrs property) (tvm.relax.op.op_attrs.ROIPoolAttrs property) specialize() (tvm.tirx.PrimFunc method) SpecializePrimFuncBasedOnCallSite() (in module tvm.relax.transform) split() (in module tvm.relax.op) (tvm.s_tir.schedule.Schedule method) SplitAttrs (class in tvm.relax.op.op_attrs) SplitCallTIRByPattern() (in module tvm.relax.transform) SplitHostDevice() (in module tvm.tirx.transform) SplitLayoutRewritePreproc() (in module tvm.relax.transform) sqrt() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) square() (in module tvm.relax.op) squeeze() (in module tvm.relax.op) SqueezeAttrs (class in tvm.relax.op.op_attrs) stack() (in module tvm.relax.op) StackAttrs (class in tvm.relax.op.op_attrs) start_checkpoint() (in module tvm.relax.op.grad) start_profile_intrinsic() (in module tvm.tirx) stat() (tvm.testing.Path method) state (tvm.s_tir.schedule.Schedule property) StaticPlanBlockMemory() (in module tvm.relax.transform) StatisticalAttrs (class in tvm.relax.op.op_attrs) stats() (tvm.relax.VMExecutable method) StatusKind (class in tvm.support.popen_pool) std() (in module tvm.relax.op) Stmt (class in tvm.tirx) stmt (tvm.s_tir.schedule.StmtSRef property) stmt_list() (in module tvm.tirx) stmt_seq() (in module tvm.tirx) StmtExprMutator (class in tvm.tirx.stmt_functor) StmtExprVisitor (class in tvm.tirx.stmt_functor) StmtFunctor (class in tvm.tirx.stmt_functor) StmtMutator (class in tvm.tirx.stmt_functor) StmtSimplify() (in module tvm.tirx.transform) StmtSimplifyConfig (class in tvm.tirx.transform) StmtSRef (class in tvm.s_tir.schedule) StmtVisitor (class in tvm.tirx.stmt_functor) stop_lift_params() (in module tvm.relax.op.builtin) storage_align() (tvm.s_tir.schedule.Schedule method) StorageRewrite() (in module tvm.tirx.transform) stride_h (tvm.topi.nn.Workload attribute) stride_w (tvm.topi.nn.Workload attribute) strided_set_python() (in module tvm.topi.testing) strided_slice() (in module tvm.relax.op) (in module tvm.topi.nn) strided_slice_python() (in module tvm.topi.testing) StridedSliceAttrs (class in tvm.relax.op.op_attrs) strides (tvm.relax.op.op_attrs.Conv1DAttrs property) (tvm.relax.op.op_attrs.Conv1DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv2DAttrs property) (tvm.relax.op.op_attrs.Conv2DTransposeAttrs property) (tvm.relax.op.op_attrs.Conv3DAttrs property) (tvm.relax.op.op_attrs.Conv3DTransposeAttrs property) (tvm.relax.op.op_attrs.Pool1DAttrs property) (tvm.relax.op.op_attrs.Pool2DAttrs property) (tvm.relax.op.op_attrs.Pool3DAttrs property) StringImm (class in tvm.relax) (class in tvm.tirx) strtobool() (in module tvm.testing) Sub (class in tvm.tirx) submit() (tvm.support.popen_pool.PopenPoolExecutor method) subscripts (tvm.relax.op.op_attrs.EinsumAttrs property) subspace_divide() (in module tvm.arith) substitute() (in module tvm.tirx.stmt_functor) subtract() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) suggest_layout_transforms() (in module tvm.relax.analysis) sum() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) summary() (tvm.rpc.TrackerSession method) support_block_scope_hoisting (tvm.tirx.transform.HoistIfThenElseConfig property) SwizzleLayout (class in tvm.tirx) symlink_to() (tvm.testing.Path method) sync_worker_0() (tvm.runtime.disco.Session method) system_lib() (tvm.rpc.RPCSession method)

T

table() (tvm.s_tir.meta_schedule.Profiler 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.tirx) tanh() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) tar() (in module tvm.support.tar) Target (class in tvm.target) target (tvm.target.VirtualDevice property) target_or_current() (tvm.target.Target static method) TargetKind (class in tvm.target) TaskScheduler (class in tvm.s_tir.meta_schedule) te_grad_kwargs (tvm.relax.op.op_attrs.CallTIRWithGradAttrs property) te_grad_name (tvm.relax.op.op_attrs.CallTIRWithGradAttrs property) tempdir() (in module tvm.support.utils) TempDirectory (class in tvm.support.utils) Tensor (class in tvm.te) tensor_to_shape() (in module tvm.relax.op) TensorCacheShardingManager (class in tvm.contrib.tvmjs) TensorIntrin (class in tvm.tirx) tensorize() (tvm.s_tir.schedule.Schedule method) TensorSlice (class in tvm.te) TensorType (class in tvm.relax) terminate() (tvm.rpc.Server method) terminate_self() (in module tvm.testing) terminate_task() (tvm.s_tir.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) TextureFlatten() (in module tvm.s_tir.backend.adreno.transform) thread_axis() (in module tvm.te) ThreadedSession (class in tvm.runtime.disco) threshold (tvm.relax.op.op_attrs.MultiboxTransformLocAttrs property) (tvm.relax.op.op_attrs.SoftplusAttrs property) tile() (in module tvm.relax.op) (tvm.tirx.Layout method) tile_to() (tvm.tirx.Layout method) TileAttrs (class in tvm.relax.op.op_attrs) TileLayout (class in tvm.tirx) TilePrimitiveCall (class in tvm.tirx) TilePrimitiveDispatch() (in module tvm.tirx.transform) time_evaluator() (tvm.relax.VirtualMachine method) (tvm.runtime.vm.VirtualMachine method) timeit() (tvm.s_tir.meta_schedule.Profiler static method) timer_end_cuda() (in module tvm.backend.cuda.op) timer_finalize_cuda() (in module tvm.backend.cuda.op) timer_init_cuda() (in module tvm.backend.cuda.op) timer_start_cuda() (in module tvm.backend.cuda.op) tir_vars_in_type() (in module tvm.relax.analysis) to_psum() (tvm.tirx.TileLayout method) 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) toolchain_version() (in module tvm.contrib.hexagon) top_k (tvm.relax.op.op_attrs.NonMaximumSuppressionAttrs property) topk() (in module tvm.relax.op) TopKAttrs (class in tvm.relax.op.op_attrs) TopologicalSort() (in module tvm.relax.transform) touch() (tvm.testing.Path method) touch_task() (tvm.s_tir.meta_schedule.TaskScheduler method) ToVDeviceAttrs (class in tvm.relax.op.op_attrs) Trace (class in tvm.s_tir.schedule) trace (tvm.s_tir.schedule.Schedule property) trace() (in module tvm.te) (in module tvm.tirx) TrackerSession (class in tvm.rpc) Trainer (class in tvm.relax.training) training (tvm.relax.op.op_attrs.BatchNormAttrs property) trainium() (tvm.tirx.TileLayout class method) transform_block_layout() (tvm.s_tir.schedule.Schedule method) transform_layout() (tvm.s_tir.schedule.Schedule method) transitively_prove_inequalities (tvm.tirx.transform.StmtSimplifyConfig property) tril() (in module tvm.relax.op) TriluAttrs (class in tvm.relax.op.op_attrs) triu() (in module tvm.relax.op) trunc() (in module tvm.relax.op) (in module tvm.te) (in module tvm.tirx) truncdiv() (in module tvm.te) (in module tvm.tirx) truncmod() (in module tvm.te) (in module tvm.tirx) try_compare() (tvm.arith.Analyzer method) try_inline() (in module tvm.s_tir.dlight) try_inline_contiguous_spatial() (in module tvm.s_tir.dlight) tune() (tvm.s_tir.meta_schedule.TaskScheduler method) tune_tasks() (in module tvm.s_tir.meta_schedule) tune_tir() (in module tvm.s_tir.meta_schedule) TuneContext (class in tvm.s_tir.meta_schedule) Tuple (class in tvm.relax) TupleGetItem (class in tvm.relax) TupleType (class in tvm.ir) tvm.arith module tvm.backend.cuda module tvm.backend.cuda.lang module tvm.backend.cuda.op module tvm.backend.cuda.operator module tvm.backend.cuda.script module tvm.backend.cuda.target_tags module tvm.contrib module tvm.contrib.cblas module tvm.contrib.coreml_runtime module tvm.contrib.cublas module tvm.contrib.cublaslt module tvm.contrib.cudnn module tvm.contrib.cutlass module tvm.contrib.dlpack module tvm.contrib.dnnl module tvm.contrib.download module tvm.contrib.hexagon module tvm.contrib.hipblas module tvm.contrib.mkl module tvm.contrib.nnpack module tvm.contrib.pickle_memoize module tvm.contrib.random module tvm.contrib.thrust module tvm.contrib.tvmjs module tvm.driver module tvm.error module tvm.exec module tvm.ir module tvm.ir.instrument module tvm.ir.transform module tvm.relax module tvm.relax.analysis module

tvm.relax.backend module tvm.relax.block_builder module tvm.relax.distributed module tvm.relax.dpl module tvm.relax.frontend module tvm.relax.frontend.onnx module tvm.relax.frontend.stablehlo module tvm.relax.frontend.tflite 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.op.vision module tvm.relax.op.vm module tvm.relax.training module tvm.relax.transform module tvm.relax.transform.legalize_ops module tvm.rpc module tvm.runtime module tvm.runtime.disco module tvm.runtime.vm module tvm.s_tir.analysis module tvm.s_tir.backend module tvm.s_tir.backend.adreno module tvm.s_tir.backend.adreno.pipeline module tvm.s_tir.backend.adreno.transform module tvm.s_tir.dlight module tvm.s_tir.meta_schedule module tvm.s_tir.schedule module tvm.s_tir.tensor_intrin module tvm.script module tvm.script.ir_builder module tvm.script.ir_builder.ir module tvm.script.ir_builder.relax module tvm.script.ir_builder.relax.distributed module tvm.script.ir_builder.tirx module tvm.script.parser module tvm.script.parser.ir module tvm.script.parser.relax module tvm.script.parser.tirx module tvm.script.printer module tvm.script.relax module tvm.script.tirx module tvm.support module tvm.support.cc module tvm.support.clang module tvm.support.emcc module tvm.support.ndk module tvm.support.nvcc module tvm.support.popen_pool module tvm.support.rocm module tvm.support.tar module tvm.support.utils module tvm.support.xcode module tvm.target module tvm.target.tag module tvm.te module tvm.testing module tvm.tirx module tvm.tirx.analysis.analysis module tvm.tirx.stmt_functor module tvm.tirx.transform module tvm.topi.cpp module tvm.topi.image module tvm.topi.nn module tvm.topi.testing module tvm_access_ptr() (in module tvm.backend.cuda.op) (in module tvm.tirx) tvm_bmma_sync() (in module tvm.tirx) tvm_fill_fragment() (in module tvm.tirx) tvm_load_matrix_sync() (in module tvm.tirx) tvm_mma_sync() (in module tvm.tirx) tvm_stack_alloca() (in module tvm.tirx) tvm_stack_make_array() (in module tvm.tirx) tvm_stack_make_shape() (in module tvm.tirx) tvm_store_matrix_sync() (in module tvm.tirx) tvm_struct_get() (in module tvm.tirx) tvm_struct_set() (in module tvm.tirx) tvm_thread_allreduce() (in module tvm.tirx) tvm_throw_last_error() (in module tvm.tirx) tvm_tuple() (in module tvm.tirx) TVMBackendAllocWorkspace() (in module tvm.tirx) TVMBackendFreeWorkspace() (in module tvm.tirx) Type (class in tvm.ir) type_annotation() (in module tvm.tirx) type_base_check() (in module tvm.relax.analysis) type_lca() (in module tvm.relax.analysis)

U

udchain() (in module tvm.relax.analysis) unannotate() (tvm.s_tir.schedule.Schedule method) undef() (in module tvm.tirx) undefined_vars() (in module tvm.tirx.analysis.analysis) uniform() (in module tvm.contrib.random) unique() (in module tvm.relax.op) unlink() (tvm.testing.Path method) unpack() (tvm.tirx.Layout method) unpack_NCHWc_to_nchw() (in module tvm.topi.nn) unroll() (tvm.s_tir.schedule.Schedule method) unroll_local_access (tvm.tirx.transform.UnrollLoopConfig property) UnrollLoop() (in module tvm.tirx.transform) UnrollLoopConfig (class in tvm.tirx.transform) unsafe_hide_buffer_access() (tvm.s_tir.schedule.Schedule method) unsafe_set_dtype() (tvm.s_tir.schedule.Schedule method) untar() (in module tvm.support.tar)

update() (tvm.arith.Analyzer method) (tvm.ir.IRModule method) (tvm.relax.training.Trainer method) (tvm.s_tir.meta_schedule.CostModel method) update_func() (tvm.ir.IRModule method) (tvm.relax.block_builder.BlockBuilder method) update_global_info() (tvm.ir.IRModule method) update_single_record() (tvm.contrib.tvmjs.TensorCacheShardingManager method) UpdateParamType() (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) upscale_factor (tvm.relax.op.op_attrs.PixelShuffleAttrs property) used_vars() (in module tvm.relax.analysis) UsingBlockVar (tvm.tirx.transform.HoistedConditionals attribute)

V

values() (tvm.ir.Map method) Var (class in tvm.relax) (class in tvm.tirx) var() (in module tvm.te) VarBinding (class in tvm.relax) variance() (in module tvm.relax.op) variances (tvm.relax.op.op_attrs.MultiboxTransformLocAttrs property) VDevice (class in tvm.ir) vdevice() (in module tvm.script.ir_builder.ir) vector() (tvm.arith.IntSet static method) vectorcombine() (in module tvm.tirx) vectorhigh() (in module tvm.tirx) vectorize() (tvm.s_tir.schedule.Schedule method) VectorizeLoop() (in module tvm.tirx.transform) vectorlow() (in module tvm.tirx) VERIFY_CACHED_FLAGS (tvm.s_tir.schedule.ScheduleDebugMask attribute) verify_gpu_code() (in module tvm.s_tir.analysis) verify_memory() (in module tvm.tirx.analysis.analysis) VERIFY_SREF_TREE (tvm.s_tir.schedule.ScheduleDebugMask attribute) verify_ssa() (in module tvm.tirx.analysis.analysis) verify_tirx_well_formed() (in module tvm.tirx.analysis.analysis) verify_well_formed() (in module tvm.tirx.analysis.analysis) (tvm.tirx.Layout method) VerifyMemory() (in module tvm.tirx.transform) view() (in module tvm.relax.op.memory) (tvm.tirx.Buffer method) virtual_device_id (tvm.target.VirtualDevice property) VirtualDevice (class in tvm.target) VirtualMachine (class in tvm.relax) (class in tvm.runtime.vm) visit_add_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_alloc_buffer_() (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_allocate_() (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_allocate_const_() (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_and_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_assert_() (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_assert_stmt_() (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_attr_() (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_attr_stmt_() (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_bind_() (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) 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_block_() (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_block_realize_() (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_break_() (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_broadcast_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_buffer_load_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_buffer_realize_() (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) visit_buffer_region_() (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_buffer_store_() (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_call_() (tvm.relax.PyExprMutator method) (tvm.relax.PyExprVisitor method) (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_cast_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_constant_() (tvm.relax.PyExprMutator method) (tvm.relax.PyExprVisitor method) visit_continue_() (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor 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_decl_buffer_() (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_div_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_eq_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_evaluate_() (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_expr() (tvm.relax.ExprFunctor method) (tvm.relax.PyExprMutator method) (tvm.relax.PyExprVisitor method) (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) (tvm.tirx.stmt_functor.StmtExprMutator method) (tvm.tirx.stmt_functor.StmtExprVisitor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_expr_default_() (tvm.tirx.ExprFunctor method) visit_expr_post_order() (tvm.relax.PyExprMutator method) visit_extern_func_() (tvm.relax.PyExprMutator method) (tvm.relax.PyExprVisitor method) visit_float_imm_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_floor_div_() (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_floor_mod_() (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method)

visit_floordiv_() (tvm.tirx.ExprFunctor method) visit_floormod_() (tvm.tirx.ExprFunctor method) visit_for_() (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_function_() (tvm.relax.PyExprMutator method) (tvm.relax.PyExprVisitor method) visit_ge_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_global_var_() (tvm.relax.PyExprMutator method) (tvm.relax.PyExprVisitor method) visit_gt_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_if_() (tvm.relax.PyExprMutator method) (tvm.relax.PyExprVisitor method) visit_if_then_else_() (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_int_imm_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_le_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_let_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_lt_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_match_cast_() (tvm.relax.PyExprMutator method) (tvm.relax.PyExprVisitor method) visit_max_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_min_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_mod_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_mul_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_ne_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_not_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_op_() (tvm.relax.PyExprMutator method) (tvm.relax.PyExprVisitor method) visit_op_call_() (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_or_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_prefetch_() (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) visit_prim_value_() (tvm.relax.PyExprMutator method) (tvm.relax.PyExprVisitor method) visit_producer_load_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_producer_realize_() (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) visit_producer_store_() (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) visit_ramp_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_reduce_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_sblock_() (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_sblock_realize_() (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_scope_id_def_stmt_() (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_select_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_seq_expr_() (tvm.relax.PyExprMutator method) (tvm.relax.PyExprVisitor method) visit_seq_stmt_() (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_seqstmt_() (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_shape_expr_() (tvm.relax.PyExprMutator method) (tvm.relax.PyExprVisitor method) visit_shuffle_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_size_var_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_span() (tvm.relax.PyExprMutator method) (tvm.relax.PyExprVisitor method) visit_stmt() (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) (tvm.tirx.stmt_functor.StmtFunctor method) visit_stmt_default_() (tvm.tirx.stmt_functor.StmtFunctor method) visit_string_imm_() (tvm.relax.PyExprMutator method) (tvm.relax.PyExprVisitor method) (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) visit_sub_() (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor 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) (tvm.tirx.ExprFunctor method) (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor 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_while_() (tvm.tirx.PyStmtExprMutator method) (tvm.tirx.PyStmtExprVisitor method) (tvm.tirx.stmt_functor.StmtFunctor method) (tvm.tirx.stmt_functor.StmtMutator method) (tvm.tirx.stmt_functor.StmtVisitor method) visit_with_new_scope() (tvm.relax.PyExprMutator method) vload() (tvm.tirx.Buffer method) VMBuiltinLower() (in module tvm.relax.transform) VMExecutable (class in tvm.relax) VMInstrumentReturnKind (class in tvm.relax) (class in tvm.runtime.vm) VMShapeLower() (in module tvm.relax.transform) vscale() (in module tvm.tirx) vstore() (tvm.tirx.Buffer method) vulkan() (tvm.rpc.RPCSession method)

W

WAR (tvm.s_tir.schedule.DepKind attribute) WAW (tvm.s_tir.schedule.DepKind attribute) webgpu() (tvm.rpc.RPCSession method) well_formed() (in module tvm.relax.analysis) where() (in module tvm.relax.op) which() (in module tvm.support.utils) While (class in tvm.tirx) width (tvm.topi.nn.Workload attribute) window_size (tvm.relax.op.op_attrs.AttentionAttrs property) winograd_transform_matrices() (in module tvm.topi.nn) with_allocated_addr() (tvm.tirx.Buffer method) with_attr() (tvm.ir.BaseFunc method) (tvm.ir.IRModule method) with_attrs() (tvm.ir.BaseFunc method) (tvm.ir.IRModule method)

with_body() (tvm.tirx.PrimFunc method) with_data() (tvm.tirx.Buffer method) with_decision() (tvm.s_tir.schedule.Trace method) with_dtype() (tvm.tirx.Buffer method) with_minrpc() (in module tvm.rpc) with_type() (tvm.relax.PyExprMutator method) with_workspace() (tvm.tirx.TilePrimitiveCall method) without_attr() (tvm.ir.BaseFunc method) (tvm.ir.IRModule method) work_on() (tvm.s_tir.schedule.Schedule method) Workload (class in tvm.topi.nn) wrap_param() (in module tvm.relax.op) WrapParamAttrs (class in tvm.relax.op.op_attrs) write_bytes() (tvm.testing.Path method) write_text() (tvm.testing.Path method)

X

xaiver_uniform_init_params() (tvm.relax.training.Trainer method)

xcrun() (in module tvm.support.xcode)

Z

zero_init_params() (tvm.relax.training.Trainer method) zero_init_states() (tvm.relax.training.Trainer method)

zeros() (in module tvm.relax.op) zeros_like() (in module tvm.relax.op)