Index
A
- abs() (in module tvm.relax.op)
- access_ptr() (tvm.tir.Buffer method)
- accumulate (tvm.relax.op.op_attrs.IndexPutAttrs property)
- acos() (in module tvm.relax.op)
- acosh() (in module tvm.relax.op)
- adaptive_avg_pool1d() (in module tvm.relax.op.nn)
- adaptive_avg_pool2d() (in module tvm.relax.op.nn)
- adaptive_avg_pool3d() (in module tvm.relax.op.nn)
- adaptive_pool() (in module tvm.topi.nn)
- adaptive_pool1d() (in module tvm.topi.nn)
- adaptive_pool3d() (in module tvm.topi.nn)
- 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.tir)
- add() (in module tvm.relax.op)
- 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.tir.schedule.Schedule method)
- address_of() (in module tvm.tir)
- AdjustMatmulOrder() (in module tvm.relax.transform)
- adreno() (in module tvm.target)
- affine_grid() (in module tvm.topi.image)
- All (tvm.tir.transform.HoistedConditionals attribute)
- all() (in module tvm.te)
- all_global_vars() (in module tvm.relax.analysis)
- all_vars() (in module tvm.relax.analysis)
- allgather() (in module tvm.relax.op.ccl)
- AllGatherAttrs (class in tvm.relax.op.op_attrs)
- alloc_storage() (in module tvm.relax.op.memory)
- alloc_tensor() (in module tvm.relax.op.builtin)
- Allocate (class in tvm.tir)
- AllocateConst (class in tvm.tir)
- AllocateWorkspace() (in module tvm.relax.transform)
- allreduce() (in module tvm.relax.op.ccl)
- AllReduceAttrs (class in tvm.relax.op.op_attrs)
- alpha (tvm.relax.op.op_attrs.LeakyReluAttrs property)
- AlterOpImpl() (in module tvm.relax.transform)
- And (class in tvm.tir)
- annotate() (tvm.tir.schedule.Schedule method)
- annotate_buffer_access() (tvm.tir.schedule.Schedule method)
- annotate_sharding() (in module tvm.relax.op.distributed)
- AnnotateDeviceRegions() (in module tvm.tir.transform)
- AnnotateEntryFunc() (in module tvm.tir.transform)
- AnnotateIrregularLoop() (in module tvm.tir.transform)
- AnnotateTIROpPattern() (in module tvm.relax.transform)
- any() (in module tvm.te)
- append() (tvm.tir.schedule.Trace method)
- Apply() (in module tvm.tir.transform)
- apply() (tvm.dlight.ScheduleRule method)
- apply_constraints_to_boolean_branches (tvm.tir.transform.SimplifyConfig property)
- apply_json_to_schedule() (tvm.tir.schedule.Trace static method)
- apply_to_schedule() (tvm.tir.schedule.Trace method)
- ApplyDefaultSchedule (class in tvm.dlight)
- ApplyPassToFunction() (in module tvm.transform)
- arange() (in module tvm.relax.op)
- arch (tvm.target.Target property)
- argmax() (in module tvm.relax.op)
- ArgmaxArgminAttrs (class in tvm.relax.op.op_attrs)
- argmin() (in module tvm.relax.op)
- argsort() (in module tvm.relax.op)
- ArgsortAttrs (class in tvm.relax.op.op_attrs)
- arm_cpu() (in module tvm.target)
- Array (class in tvm.ir)
- as_json() (tvm.tir.schedule.Trace method)
- as_python() (tvm.relax.VMExecutable method)
- as_text() (tvm.relax.VMExecutable method)
- asin() (in module tvm.relax.op)
- asinh() (in module tvm.relax.op)
- asobject() (tvm.te.TensorSlice method)
- assert_op() (in module tvm.relax.op)
- assert_pure_function() (in module tvm.tir.analysis.analysis)
- assert_structural_equal() (in module tvm.ir)
- AssertStmt (class in tvm.tir)
- assume() (in module tvm.tir)
- 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)
- atan2() (in module tvm.tir)
- atanh() (in module tvm.relax.op)
- 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_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)
- AttrStmt (class in tvm.tir)
- auto_max_depth (tvm.tir.transform.UnrollLoopConfig property)
- auto_max_extent (tvm.tir.transform.UnrollLoopConfig property)
- auto_max_step (tvm.tir.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)
- 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
- backward_index() (tvm.tir.BijectiveLayout method)
- backward_shape() (tvm.tir.BijectiveLayout method)
- BaseCheckResult (class in tvm.relax.analysis)
- BaseExpr (class in tvm.ir)
- BaseFunc (class in tvm.ir)
- BasePyModule (class in tvm.relax)
- batch_dims (tvm.relax.op.op_attrs.GatherNDAttrs property)
- batch_matmul() (in module tvm.contrib.cblas)
- batch_norm() (in module tvm.relax.op.nn)
- batch_to_space_nd() (in module tvm.topi.nn)
- BatchNormAttrs (class in tvm.relax.op.op_attrs)
- begin_scope() (tvm.relax.block_builder.BlockBuilder method)
- beta (tvm.relax.op.op_attrs.SoftplusAttrs property)
- BF16ComputeLegalize() (in module tvm.tir.transform)
- BF16StorageLegalize() (in module tvm.tir.transform)
- bifrost() (in module tvm.target)
- bijective_layout() (in module tvm.tir)
- BijectiveLayout (class in tvm.tir)
- binarize_pack() (in module tvm.topi.nn)
- binary_dense() (in module tvm.topi.nn)
- bind() (tvm.tir.schedule.Schedule method)
- bind_params() (tvm.relax.Function method)
- bind_symbolic_vars() (tvm.relax.Function method)
- Binding (class in tvm.relax)
- BindingBlock (class in tvm.relax)
- BindParams() (in module tvm.relax.transform)
- BindSymbolicVars() (in module tvm.relax.transform)
- BindTarget() (in module tvm.tir.transform)
- bitpack() (in module tvm.topi.nn)
- bitserial_conv2d_nchw() (in module tvm.topi.nn)
- bitserial_conv2d_nhwc() (in module tvm.topi.nn)
- bitserial_dense() (in module tvm.topi.nn)
- bitwise_and() (in module tvm.relax.op)
- bitwise_not() (in module tvm.relax.op)
- bitwise_or() (in module tvm.relax.op)
- bitwise_xor() (in module tvm.relax.op)
- Block (class in tvm.tir)
- BlockBuilder (class in tvm.relax.block_builder)
- BlockDependenceInfo (class in tvm.tir)
- BlockInfo (class in tvm.dlight)
- blockize() (tvm.tir.schedule.Schedule method)
- BlockRealize (class in tvm.tir)
- BlockRV (class in tvm.tir.schedule)
- BlockScope (class in tvm.tir.schedule)
- BooleanExpression (tvm.tir.transform.HoistedConditionals attribute)
- bound_vars() (in module tvm.relax.analysis)
- break_loop() (in module tvm.tir)
- Broadcast (class in tvm.tir)
- broadcast() (tvm.runtime.disco.Session method)
- broadcast_from_worker0() (in module tvm.relax.op.ccl)
- broadcast_to() (in module tvm.relax.op)
- bucketize() (in module tvm.relax.op)
- Buffer (class in tvm.tir)
- BufferLoad (class in tvm.tir)
- BufferRealize (class in tvm.tir)
- BufferRegion (class in tvm.tir)
- BufferStore (class in tvm.tir)
- build() (in module tvm.relax)
- Builder (class in tvm.meta_schedule)
- BundleModelParams() (in module tvm.relax.transform)
C
- c() (tvm.relax.ExecBuilder method)
- Cache (class in tvm.contrib.pickle_memoize)
- cache (tvm.contrib.pickle_memoize.Cache property)
- cache_index() (tvm.tir.schedule.Schedule method)
- cache_inplace() (tvm.tir.schedule.Schedule method)
- cache_read() (tvm.tir.schedule.Schedule method)
- cache_write() (tvm.tir.schedule.Schedule method)
- calculate_allocated_bytes() (in module tvm.tir.analysis.analysis)
- Call (class in tvm.relax)
- call_builtin_with_ctx() (in module tvm.relax.op)
- call_cpacked() (in module tvm.tir)
- call_cpacked_lowered() (in module tvm.tir)
- call_dps_packed() (in module tvm.relax)
- call_extern() (in module tvm.tir)
- call_inplace_packed() (in module tvm.relax.op)
- call_intrin() (in module tvm.tir)
- call_llvm_intrin() (in module tvm.tir)
- call_llvm_pure_intrin() (in module tvm.tir)
- call_packed() (in module tvm.tir)
- call_packed_lowered() (in module tvm.tir)
- call_pure_extern() (in module tvm.tir)
- call_pure_packed() (in module tvm.relax)
- call_py_func() (in module tvm.relax.op)
- call_te() (tvm.relax.block_builder.BlockBuilder method)
- call_te_with_grad() (tvm.relax.block_builder.BlockBuilder method)
- call_tir() (in module tvm.relax)
- call_tir_inplace() (in module tvm.relax)
- call_tir_local_view() (in module tvm.relax.op.distributed)
- call_tir_with_grad() (in module tvm.relax)
- CallEffectKind (class in tvm.tir)
- CallingConv (class in tvm.ir)
- CallInplacePackedAttrs (class in tvm.relax.op.op_attrs)
- calls (tvm.runtime.profiling.Report attribute)
- 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.tir.schedule.Schedule method)
- canon_multi_target() (tvm.target.Target static method)
- canon_multi_target_and_host() (tvm.target.Target static method)
- canon_target() (tvm.target.Target static method)
- canon_target_and_host() (tvm.target.Target static method)
- canon_target_map_and_host() (tvm.target.Target static method)
- CanonicalizeBindings() (in module tvm.relax.transform)
- Cast (class in tvm.tir)
- causal_mask (tvm.relax.op.op_attrs.AttentionAttrs property)
- ceil() (in module tvm.relax.op)
- ceil_mode (tvm.relax.op.op_attrs.Pool1DAttrs property)
- ceildiv() (in module tvm.tir)
- center (tvm.relax.op.op_attrs.BatchNormAttrs property)
- channel_axis (tvm.relax.op.op_attrs.GroupNormAttrs property)
- circular_pad() (in module tvm.topi.nn)
- cl() (tvm.rpc.RPCSession method)
- clip() (in module tvm.relax.op)
- clone() (tvm.meta_schedule.Mutator method)
- close() (tvm.rpc.TrackerSession method)
- clz() (in module tvm.tir)
- collapse_sum_like() (in module tvm.relax.op)
- collapse_sum_to() (in module tvm.relax.op)
- collect_non_negative_expressions() (in module tvm.relax.analysis)
- CombineContextCall() (in module tvm.tir.transform)
- CombineParallelMatmul() (in module tvm.relax.transform)
- comm_reducer() (in module tvm.te)
- commit_tuning_record() (tvm.meta_schedule.Database method)
- commit_workload() (tvm.meta_schedule.Database method)
- CommonSubexprElimTIR() (in module tvm.tir.transform)
- CommReducer (class in tvm.tir)
- CompactBufferAllocation() (in module tvm.tir.transform)
- compile() (in module tvm)
- compile_coreml() (in module tvm.contrib.xcode)
- compile_cuda() (in module tvm.contrib.nvcc)
- compile_metal() (in module tvm.contrib.xcode)
- computable_at_compile_time() (in module tvm.relax.analysis)
- compute() (in module tvm.te)
- compute_at() (tvm.tir.schedule.Schedule method)
- compute_inline() (tvm.tir.schedule.Schedule method)
- ComputeOp (class in tvm.te)
- ComputePrimValue() (in module tvm.relax.transform)
- concat() (in module tvm.relax.op)
- ConcatAttrs (class in tvm.relax.op.op_attrs)
- concatenate() (in module tvm.topi.nn)
- connect() (in module tvm.rpc)
- connect_tracker() (in module tvm.rpc)
- const() (in module tvm.relax)
- Constant (class in tvm.relax)
- contains_impure_call() (in module tvm.relax.analysis)
- continue_loop() (in module tvm.tir)
- conv() (in module tvm.topi.nn)
- conv1d() (in module tvm.relax.op.nn)
- conv1d_ncw() (in module tvm.topi.nn)
- conv1d_nwc() (in module tvm.topi.nn)
- conv1d_transpose() (in module tvm.relax.op.nn)
- conv1d_transpose_ncw() (in module tvm.topi.nn)
- Conv1DAttrs (class in tvm.relax.op.op_attrs)
- Conv1DTransposeAttrs (class in tvm.relax.op.op_attrs)
- conv2d() (in module tvm.relax.op.nn)
- conv2d_forward() (in module tvm.contrib.miopen)
- conv2d_hwcn() (in module tvm.topi.nn)
- conv2d_nchw() (in module tvm.topi.nn)
- conv2d_NCHWc() (in module tvm.topi.nn)
- conv2d_NCHWc_int8() (in module tvm.topi.nn)
- conv2d_nhwc() (in module tvm.topi.nn)
- conv2d_transpose() (in module tvm.relax.op.nn)
- conv2d_transpose_nchw() (in module tvm.topi.nn)
- conv2d_transpose_nchw_preprocess() (in module tvm.topi.nn)
- conv2d_winograd_nchw() (in module tvm.topi.nn)
- conv2d_winograd_nchw_without_weight_transform() (in module tvm.topi.nn)
- conv2d_winograd_nhwc() (in module tvm.topi.nn)
- conv2d_winograd_nhwc_without_weight_transform() (in module tvm.topi.nn)
- conv2d_winograd_weight_transform() (in module tvm.topi.nn)
- Conv2DAttrs (class in tvm.relax.op.op_attrs)
- Conv2DTransposeAttrs (class in tvm.relax.op.op_attrs)
- conv3d() (in module tvm.relax.op.nn)
- conv3d_ncdhw() (in module tvm.topi.nn)
- conv3d_ndhwc() (in module tvm.topi.nn)
- conv3d_transpose_ncdhw() (in module tvm.topi.nn)
- conv3d_transpose_ncdhw_preprocess() (in module tvm.topi.nn)
- conv3d_winograd_weight_transform() (in module tvm.topi.nn)
- Conv3DAttrs (class in tvm.relax.op.op_attrs)
- convert_boolean_to_and_of_ors (tvm.tir.transform.SimplifyConfig property)
- convert_func() (in module tvm.contrib.dlpack)
- convert_to_expr() (in module tvm.relax)
- ConvertBlocksToOpaque() (in module tvm.tir.transform)
- ConvertForLoopsToSerial() (in module tvm.tir.transform)
- ConvertLayout() (in module tvm.relax.transform)
- ConvertSSA() (in module tvm.tir.transform)
- ConvertToDataflow() (in module tvm.relax.transform)
- convolution_inference() (in module tvm.contrib.nnpack)
- convolution_inference_weight_transform() (in module tvm.contrib.nnpack)
- convolution_inference_without_weight_transform() (in module tvm.contrib.nnpack)
- coordinate_transformation_mode (tvm.relax.op.op_attrs.Resize2DAttrs property)
- copy() (tvm.tir.schedule.Schedule method)
- copy_from_worker_0() (tvm.runtime.disco.Session method)
- copy_to_worker_0() (tvm.runtime.disco.Session method)
- copysign() (in module tvm.tir)
- correlation_nchw() (in module tvm.topi.nn)
- cos() (in module tvm.relax.op)
- cosh() (in module tvm.relax.op)
- CostModel (class in tvm.meta_schedule)
- Count (class in tvm.runtime.profiling)
- count_include_pad (tvm.relax.op.op_attrs.Pool1DAttrs property)
- cpu() (tvm.rpc.RPCSession method)
- create() (in module tvm.target)
- (tvm.meta_schedule.Builder static method)
- (tvm.meta_schedule.CostModel static method)
- (tvm.meta_schedule.Database static method)
- (tvm.meta_schedule.FeatureExtractor static method)
- (tvm.meta_schedule.MeasureCallback static method)
- (tvm.meta_schedule.Mutator static method)
- (tvm.meta_schedule.Postproc static method)
- (tvm.meta_schedule.Runner static method)
- (tvm.meta_schedule.ScheduleRule static method)
- (tvm.meta_schedule.SearchStrategy static method)
- (tvm.meta_schedule.SpaceGenerator static method)
- (tvm.meta_schedule.TaskScheduler static method)
- create_barriers() (in module tvm.tir)
- create_dylib() (in module tvm.contrib.xcode)
- create_empty() (tvm.relax.Function static method)
- create_executable() (in module tvm.contrib.cc)
- create_llvm() (in module tvm.contrib.clang)
- create_prim_func() (in module tvm.te)
- create_shared() (in module tvm.contrib.cc)
- create_staticlib() (in module tvm.contrib.cc)
- create_tvmjs_wasm() (in module tvm.contrib.emcc)
- crop_and_resize() (in module tvm.topi.image)
- cross_compiler() (in module tvm.contrib.cc)
- cross_entropy_with_logits() (in module tvm.relax.op.nn)
- csv() (tvm.runtime.profiling.Report method)
- cubic_alpha (tvm.relax.op.op_attrs.Resize2DAttrs property)
- cubic_exclude (tvm.relax.op.op_attrs.Resize2DAttrs property)
- cuda() (in module tvm.target)
- cumprod() (in module tvm.relax.op)
- cumsum() (in module tvm.relax.op)
- current() (tvm.meta_schedule.Database static method)
- current_block_is_dataflow() (tvm.relax.block_builder.BlockBuilder method)
D
- data_layout (tvm.relax.op.op_attrs.Conv1DAttrs property)
- Database (class in tvm.meta_schedule)
- dataflow() (tvm.relax.block_builder.BlockBuilder method)
- DataflowBlock (class in tvm.relax)
- dataflowblock_pass() (in module tvm.relax.transform)
- DataflowBlockPass (class in tvm.relax.transform)
- DataflowBlockRewrite (class in tvm.relax)
- DataflowScope (class in tvm.relax.block_builder)
- DataflowUseInplaceCalls() (in module tvm.relax.transform)
- DataflowVar (class in tvm.relax)
- DataProducer (class in tvm.tir)
- DataTypeImm (class in tvm.relax)
- DeadCodeElimination() (in module tvm.relax.transform)
- debug_copy_from() (tvm.runtime.disco.DRef method)
- debug_get_from_remote() (tvm.runtime.disco.DRef method)
- decisions (tvm.tir.schedule.Trace attribute)
- decl_buffer() (in module tvm.tir)
- declaration_conv2d_transpose_impl() (in module tvm.topi.nn)
- declaration_conv3d_transpose_impl() (in module tvm.topi.nn)
- declare_function() (tvm.relax.ExecBuilder method)
- DeclBuffer (class in tvm.tir)
- decompose_padding() (tvm.tir.schedule.Schedule method)
- decompose_reduction() (tvm.tir.schedule.Schedule method)
- DecomposeOpsForInference() (in module tvm.relax.transform)
- DecomposeOpsForTraining() (in module tvm.relax.transform)
- DecorateDeviceScope() (in module tvm.tir.transform)
- DefaultGPUSchedule() (in module tvm.tir.transform)
- definable_tir_vars_in_struct_info() (in module tvm.relax.analysis)
- defined_symbolic_vars() (in module tvm.relax.analysis)
- deformable_conv2d_nchw() (in module tvm.topi.nn)
- deformable_conv2d_nhwc() (in module tvm.topi.nn)
- dense() (in module tvm.topi.nn)
- dense_pack() (in module tvm.topi.nn)
- Dependency (class in tvm.tir.schedule)
- DepKind (class in tvm.tir.schedule)
- depth (tvm.relax.op.op_attrs.OneHotAttrs property)
- depth_to_space() (in module tvm.topi.nn)
- depthwise_conv2d_backward_input_nhwc() (in module tvm.topi.nn)
- depthwise_conv2d_backward_weight_nhwc() (in module tvm.topi.nn)
- depthwise_conv2d_nchw() (in module tvm.topi.nn)
- depthwise_conv2d_NCHWc() (in module tvm.topi.nn)
- depthwise_conv2d_nhwc() (in module tvm.topi.nn)
- dequantize() (in module tvm.relax.op)
- derive_call_ret_struct_info() (in module tvm.relax.analysis)
- derived_object() (in module tvm.meta_schedule)
- descending (tvm.relax.op.op_attrs.ArgsortAttrs property)
- detach_params() (in module tvm.relax.frontend)
- detect_buffer_access_lca() (in module tvm.tir.analysis.analysis)
- detect_recursion() (in module tvm.relax.analysis)
- device() (tvm.rpc.RPCSession method)
- device_metrics (tvm.runtime.profiling.Report attribute)
- device_type (tvm.relax.op.op_attrs.HintOnDeviceAttrs property)
- device_type_int (tvm.target.VirtualDevice property)
- DeviceWrapper (class in tvm.runtime.profiling)
- DiagnosticError
- DictAttrs (class in tvm.ir)
- dilate() (in module tvm.topi.nn)
- 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.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)
- DirectoryCreatedPastAtExit
- Div (class in tvm.tir)
- div() (in module tvm.te)
- divide() (in module tvm.relax.op)
- DModule (class in tvm.runtime.disco)
- dom (tvm.dlight.IterInfo property)
- dom() (tvm.dlight.BlockInfo method)
- dom_kind() (tvm.dlight.BlockInfo method)
- download() (tvm.rpc.RPCSession method)
- download_linked_module() (tvm.rpc.RPCSession method)
- dp4a() (in module tvm.tir)
- DPackedFunc (class in tvm.runtime.disco)
- DRef (class in tvm.runtime.disco)
- dropout() (in module tvm.relax.op.nn)
- DropoutAttrs (class in tvm.relax.op.op_attrs)
- dst_vdevice (tvm.relax.op.op_attrs.ToVDeviceAttrs property)
- 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)
- DummyGlobalInfo (class in tvm.ir)
- dump_pruned() (tvm.meta_schedule.Database method)
- Duration (class in tvm.runtime.profiling)
- dynamic_strided_slice() (in module tvm.relax.op)
- dynamo_capture_subgraphs() (in module tvm.relax.frontend.torch)
E
- einsum() (in module tvm.relax.op)
- EinsumAttrs (class in tvm.relax.op.op_attrs)
- EliminateCommonSubexpr() (in module tvm.relax.transform)
- emit() (tvm.relax.block_builder.BlockBuilder method)
- emit_call() (tvm.relax.ExecBuilder method)
- emit_func_output() (tvm.relax.block_builder.BlockBuilder method)
- emit_goto() (tvm.relax.ExecBuilder method)
- emit_if() (tvm.relax.ExecBuilder method)
- emit_normalized() (tvm.relax.block_builder.BlockBuilder method)
- emit_output() (tvm.relax.block_builder.BlockBuilder method)
- emit_ret() (tvm.relax.ExecBuilder method)
- emit_te() (tvm.relax.block_builder.BlockBuilder method)
- empty() (tvm.runtime.disco.Session method)
- end_checkpoint() (in module tvm.relax.op.grad)
- end_profile_intrinsic() (in module tvm.tir)
- end_scope() (tvm.relax.block_builder.BlockBuilder method)
- ensure_zero_offset() (in module tvm.relax.op.memory)
- enter_pass_ctx() (tvm.instrument.PassInstrument method)
- enter_postproc() (tvm.tir.schedule.Schedule method)
- EnvFunc (class in tvm.ir)
- epsilon (tvm.relax.op.op_attrs.BatchNormAttrs property)
- EQ (class in tvm.tir)
- equal() (in module tvm.relax.op)
- equal_const_int() (in module tvm.topi.nn)
- erase_to_well_defined() (in module tvm.relax.analysis)
- erf() (in module tvm.relax.op)
- estimate_memory_usage() (in module tvm.relax.analysis)
- estimate_tir_flops() (in module tvm.tir.analysis.analysis)
- Evaluate (class in tvm.tir)
- ewise_fma() (in module tvm.relax.op)
- exclusive (tvm.relax.op.op_attrs.ScanopAttrs property)
- ExecBuilder (class in tvm.relax)
- exit_pass_ctx() (tvm.instrument.PassInstrument method)
- exp() (in module tvm.relax.op)
- exp10() (in module tvm.tir)
- exp2() (in module tvm.tir)
- expand_dims() (in module tvm.relax.op)
- ExpandDimsAttrs (class in tvm.relax.op.op_attrs)
- ExpandMatmulOfSum() (in module tvm.relax.transform)
- ExpandTupleArguments() (in module tvm.relax.transform)
- explicit_unroll (tvm.tir.transform.UnrollLoopConfig property)
- Expr (in module tvm.relax)
- expr_deep_equal() (in module tvm.tir.analysis.analysis)
- ExprFunctor (class in tvm.relax)
- ExprRV (in module tvm.tir.schedule)
- ext_dev() (tvm.rpc.RPCSession method)
- extern() (in module tvm.relax)
- extern_primfunc() (in module tvm.te)
- ExternFunc (class in tvm.relax)
- ExternOp (class in tvm.te)
- extract_from() (tvm.meta_schedule.FeatureExtractor method)
- ExtractedTask (class in tvm.meta_schedule)
- ExtractPrimFuncConstants() (in module tvm.tir.transform)
- 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)
- factor_of() (tvm.tir.Layout method)
- fast_softmax() (in module tvm.topi.nn)
- FastMathTransform (class in tvm.relax.transform)
- FeatureExtractor (class in tvm.meta_schedule)
- FewShotTuning() (in module tvm.relax.transform)
- fifo_buffer() (in module tvm.topi.nn)
- FileLock (class in tvm.contrib.utils)
- filelock() (in module tvm.contrib.utils)
- Filter() (in module tvm.tir.transform)
- finalize() (tvm.relax.block_builder.BlockBuilder method)
- find_anchor_block() (in module tvm.tir.analysis.analysis)
- find_clang() (in module tvm.contrib.clang)
- find_cuda_path() (in module tvm.contrib.nvcc)
- find_lld() (in module tvm.contrib.rocm)
- find_nvshmem_paths() (in module tvm.contrib.nvcc)
- find_rocm_path() (in module tvm.contrib.rocm)
- flatten() (in module tvm.relax.op)
- FlattenBuffer() (in module tvm.tir.transform)
- flip() (in module tvm.relax.op)
- FlipAttrs (class in tvm.relax.op.op_attrs)
- FloatImm (class in tvm.tir)
- floor() (in module tvm.relax.op)
- floor_divide() (in module tvm.relax.op)
- floor_mod() (in module tvm.relax.op)
- FloorDiv (class in tvm.tir)
- floordiv() (in module tvm.te)
- FloorMod (class in tvm.tir)
- floormod() (in module tvm.te)
- fmod() (in module tvm.te)
- FoldBatchnormToConv2D (class in tvm.relax.transform)
- FoldConstant() (in module tvm.relax.transform)
- For (class in tvm.tir)
- ForceNarrowIndexToInt32() (in module tvm.tir.transform)
- fork_seed() (tvm.tir.schedule.Schedule method)
- ForKind (class in tvm.tir)
- forward_index() (tvm.tir.BijectiveLayout method)
- forward_shape() (tvm.tir.BijectiveLayout method)
- FP8ComputeLegalize() (in module tvm.tir.transform)
- FP8StorageLegalize() (in module tvm.tir.transform)
- free_symbolic_vars() (in module tvm.relax.analysis)
- free_vars() (in module tvm.relax.analysis)
- from_callable() (tvm.dlight.ScheduleRule static method)
- from_device() (tvm.target.Target static method)
- from_exported_program() (in module tvm.relax.frontend.torch)
- from_expr() (tvm.ir.IRModule static method)
- from_func() (tvm.tir.IndexMap static method)
- from_func_with_separators() (tvm.tir.IndexMap static method)
- from_fx() (in module tvm.relax.frontend.torch)
- from_json() (tvm.runtime.profiling.Report class method)
- from_min_extent() (tvm.ir.Range static method)
- from_onnx() (in module tvm.relax.frontend.onnx)
- from_stablehlo() (in module tvm.relax.frontend.stablehlo)
- full() (in module tvm.relax.op)
- full_like() (in module tvm.relax.op)
- fully_connected_inference() (in module tvm.contrib.nnpack)
- func_working_on (tvm.tir.schedule.Schedule property)
- FuncStructInfo (class in tvm.relax)
- Function (class in tvm.relax)
- function() (tvm.relax.block_builder.BlockBuilder method)
- function_pass() (in module tvm.relax.transform)
- FunctionPass (class in tvm.relax.transform)
- functions_items() (tvm.ir.IRModule method)
- FunctionScope (class in tvm.relax.block_builder)
- FuncType (class in tvm.ir)
- fuse() (tvm.tir.schedule.Schedule method)
- FuseOps() (in module tvm.relax.transform)
- FuseOpsByPattern() (in module tvm.relax.transform)
- FuseTIR() (in module tvm.relax.transform)
- FuseTransposeMatmul (class in tvm.relax.transform)
- FusionPattern (class in tvm.relax.transform)
G
- gather_elements() (in module tvm.relax.op)
- gather_nd() (in module tvm.relax.op)
- gather_to_worker0() (tvm.runtime.disco.Session method)
- GatherElementsAttrs (class in tvm.relax.op.op_attrs)
- GatherNDAttrs (class in tvm.relax.op.op_attrs)
- GE (class in tvm.tir)
- gelu() (in module tvm.relax.op.nn)
- gelu_tanh() (in module tvm.relax.op.nn)
- generate_design_space() (tvm.meta_schedule.SpaceGenerator method)
- generate_measure_candidates() (tvm.meta_schedule.SearchStrategy method)
- get() (tvm.ir.DictAttrs method)
- (tvm.ir.EnvFunc static method)
- (tvm.ir.Map method)
- (tvm.ir.Op static method)
- (tvm.meta_schedule.Profiler method)
- (tvm.relax.block_builder.BlockBuilder method)
- (tvm.relax.ExecBuilder method)
- (tvm.tir.schedule.InstructionKind static method)
- (tvm.tir.schedule.Schedule method)
- (tvm.tir.TensorIntrin static method)
- get_1d_indices() (in module tvm.topi.image)
- get_1d_pixel() (in module tvm.topi.image)
- get_2d_indices() (in module tvm.topi.image)
- get_2d_pixel() (in module tvm.topi.image)
- get_3d_indices() (in module tvm.topi.image)
- get_3d_pixel() (in module tvm.topi.image)
- get_active_lane_mask() (in module tvm.tir)
- get_all_tuning_records() (tvm.meta_schedule.Database method)
- get_attr() (tvm.ir.IRModule method)
- get_block() (tvm.tir.schedule.Schedule method)
- get_block_access_region() (in module tvm.tir.analysis.analysis)
- get_block_read_write_region() (in module tvm.tir.analysis.analysis)
- get_block_scope() (tvm.tir.BlockDependenceInfo method)
- get_cc() (in module tvm.contrib.cc)
- get_child_blocks() (tvm.tir.schedule.Schedule method)
- get_closest_index() (in module tvm.topi.image)
- get_const_int() (in module tvm.topi.nn)
- get_const_tuple() (in module tvm.topi.nn)
- get_consumers() (tvm.tir.schedule.Schedule method)
- get_cuda_version() (in module tvm.contrib.nvcc)
- get_decision() (tvm.tir.schedule.Trace method)
- get_default_pipeline() (in module tvm.relax)
- get_default_tir_pipeline() (in module tvm.tir)
- get_deps_by_dst() (tvm.tir.schedule.BlockScope method)
- get_deps_by_src() (tvm.tir.schedule.BlockScope method)
- get_flattened_buffer() (tvm.tir.Buffer method)
- get_function() (tvm.relax.BasePyModule method)
- get_global_func() (tvm.runtime.disco.Session method)
- get_global_symbol_section_map() (in module tvm.contrib.cc)
- get_global_var() (tvm.ir.IRModule method)
- get_global_vars() (tvm.ir.IRModule method)
- get_int() (tvm.ir.Attrs method)
- get_int_tuple() (tvm.ir.Attrs method)
- get_inx() (in module tvm.topi.image)
- get_kind_attr() (tvm.target.Target method)
- get_loops() (tvm.tir.schedule.Schedule method)
- get_output_blocks() (tvm.tir.schedule.Schedule method)
- get_outputs() (tvm.relax.VirtualMachine method)
- get_pad_tuple() (in module tvm.topi.image)
- 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_pipeline() (in module tvm.relax)
- get_producers() (tvm.tir.schedule.Schedule method)
- get_shape_of() (in module tvm.relax)
- get_sref() (tvm.tir.BlockDependenceInfo method)
- get_static_type() (in module tvm.relax.analysis)
- get_str() (tvm.ir.Attrs method)
- get_target_by_dump_machine() (in module tvm.contrib.cc)
- get_target_device_type() (tvm.target.Target method)
- get_tir_pipeline() (in module tvm.tir)
- get_top_k() (tvm.meta_schedule.Database method)
- get_unique_name() (tvm.relax.block_builder.BlockBuilder method)
- get_var2val() (in module tvm.relax.analysis)
- get_var_remap() (tvm.relax.PyExprMutator method)
- get_vscale_expr() (in module tvm.tir)
- get_vtcm_compaction_passes() (in module tvm.tir.analysis.analysis)
- global_pool() (in module tvm.topi.nn)
- GlobalInfo (class in tvm.ir)
- GlobalVar (class in tvm.ir)
- Gradient() (in module tvm.relax.transform)
- greater() (in module tvm.relax.op)
- greater_equal() (in module tvm.relax.op)
- grid_sample() (in module tvm.topi.image)
- group_conv1d_ncw() (in module tvm.topi.nn)
- group_conv1d_nwc() (in module tvm.topi.nn)
- group_conv1d_transpose_ncw() (in module tvm.topi.nn)
- group_conv2d_nchw() (in module tvm.topi.nn)
- group_conv2d_nhwc() (in module tvm.topi.nn)
- group_conv2d_transpose_nchw() (in module tvm.topi.nn)
- group_conv3d_transpose_ncdhw() (in module tvm.topi.nn)
- group_norm() (in module tvm.relax.op.nn)
- GroupNormAttrs (class in tvm.relax.op.op_attrs)
- groups (tvm.relax.op.op_attrs.Conv1DAttrs property)
- GT (class in tvm.tir)
H
- hamming_window() (in module tvm.relax.op)
- handle_add_byte_offset() (in module tvm.tir)
- has_attr() (tvm.ir.Op method)
- has_reshape_pattern() (in module tvm.relax.analysis)
- has_workload() (tvm.meta_schedule.Database method)
- have_cudagraph() (in module tvm.contrib.nvcc)
- have_fp16() (in module tvm.contrib.nvcc)
- have_int8() (in module tvm.contrib.nvcc)
- have_matrixcore() (in module tvm.contrib.rocm)
- have_tensorcore() (in module tvm.contrib.nvcc)
- height (tvm.topi.nn.Workload attribute)
- hexagon() (in module tvm.target)
- hint_on_device() (in module tvm.relax.op)
- HintOnDeviceAttrs (class in tvm.relax.op.op_attrs)
- hoisted_conditionals (tvm.tir.transform.HoistExpressionConfig property)
- hoisted_let_bindings (tvm.tir.transform.HoistExpressionConfig property)
- HoistedConditionals (class in tvm.tir.transform)
- HoistedLetBindings (class in tvm.tir.transform)
- HoistExpression() (in module tvm.tir.transform)
- HoistExpressionConfig (class in tvm.tir.transform)
- HoistIfThenElse() (in module tvm.tir.transform)
- HoistIfThenElseConfig (class in tvm.tir.transform)
- hypot() (in module tvm.tir)
I
- Id (class in tvm.relax)
- If (class in tvm.relax)
- if_then_else() (in module tvm.te)
- IfElseExpr (tvm.tir.transform.HoistedConditionals attribute)
- IfElseStmt (tvm.tir.transform.HoistedConditionals attribute)
- IfThenElse (class in tvm.tir)
- ignore_index (tvm.relax.op.op_attrs.NLLLossAttrs property)
- ignore_loop_partition() (in module tvm.tir)
- imm() (tvm.relax.ExecBuilder method)
- import_python_module() (tvm.runtime.disco.Session method)
- in_dtype (tvm.topi.nn.Workload attribute)
- in_filter (tvm.topi.nn.Workload attribute)
- in_group (tvm.relax.op.op_attrs.AllGatherAttrs property)
- index (tvm.relax.op.op_attrs.HintOnDeviceAttrs property)
- index_map (tvm.relax.op.op_attrs.LayoutTransformAttrs property)
- index_of() (tvm.tir.Layout method)
- index_put() (in module tvm.relax.op)
- index_tensor() (in module tvm.relax.op)
- indexdiv() (in module tvm.te)
- indexing (tvm.relax.op.op_attrs.MeshgridAttrs property)
- IndexMap (class in tvm.tir)
- indexmod() (in module tvm.te)
- IndexPutAttrs (class in tvm.relax.op.op_attrs)
- indices_or_sections (tvm.relax.op.op_attrs.SplitAttrs property)
- InferFragment() (in module tvm.tir.transform)
- infinity() (in module tvm.tir)
- info (tvm.transform.Pass property)
- init_ccl() (tvm.runtime.disco.Session method)
- InitAttrs (class in tvm.relax.op.op_attrs)
- InjectDoubleBuffer() (in module tvm.tir.transform)
- InjectDoubleBufferConfig (class in tvm.tir.transform)
- InjectPermutedLayout() (in module tvm.tir.transform)
- InjectPTXAsyncCopy() (in module tvm.tir.transform)
- InjectPTXLDG32() (in module tvm.tir.transform)
- InjectRollingBuffer() (in module tvm.tir.transform)
- InjectSoftwarePipeline() (in module tvm.tir.transform)
- InjectVirtualThread() (in module tvm.tir.transform)
- inline_mark() (tvm.tir.schedule.StmtSRef static method)
- InlinePrivateFunctions() (in module tvm.relax.transform)
- input_axis_separators (tvm.relax.op.op_attrs.LayoutTransformAttrs property)
- inputs (tvm.tir.schedule.Instruction attribute)
- instance_norm() (in module tvm.relax.op.nn)
- InstanceNormAttrs (class in tvm.relax.op.op_attrs)
- Instruction (class in tvm.tir.schedule)
- InstructionKind (class in tvm.tir.schedule)
- InstrumentBoundCheckers() (in module tvm.tir.transform)
- InstrumentProfileIntrinsics() (in module tvm.tir.transform)
- insts (tvm.tir.schedule.Trace attribute)
- intel_graphics() (in module tvm.target)
- InternalError
- IntImm (class in tvm.tir)
- inverse() (tvm.tir.IndexMap method)
- invoke_closure() (in module tvm.relax.op)
- invoke_pure_closure() (in module tvm.relax.op)
- invoke_stateful() (tvm.relax.VirtualMachine method)
- IPCAllReduceRewrite (class in tvm.relax.transform)
- ir_transform() (in module tvm.tir.stmt_functor)
- IRModule (class in tvm.ir)
- is_available() (in module tvm.contrib.nnpack)
- is_base_of() (tvm.relax.StructInfo method)
- is_elementwise() (tvm.dlight.BlockInfo method)
- is_equivalent_to() (tvm.tir.IndexMap method)
- is_gemm() (tvm.dlight.BlockInfo method)
- is_gemv() (tvm.dlight.BlockInfo method)
- is_injective() (tvm.dlight.BlockInfo method)
- is_pure (tvm.tir.schedule.InstructionKind property)
- is_pure_function() (in module tvm.tir.analysis.analysis)
- is_reduction() (tvm.dlight.BlockInfo method)
- is_source_path() (in module tvm.contrib.utils)
- is_target_available() (tvm.dlight.ScheduleRule method)
- isfinite() (in module tvm.relax.op)
- isinf() (in module tvm.relax.op)
- isnan() (in module tvm.relax.op)
- isnullptr() (in module tvm.tir)
- items() (tvm.ir.DictAttrs method)
- IterInfo (class in tvm.dlight)
- IterVar (class in tvm.tir)
J
- join_running_task() (tvm.meta_schedule.TaskScheduler method)
- json() (tvm.runtime.profiling.Report method)
K
- k (tvm.relax.op.op_attrs.TopKAttrs property)
- keepdims (tvm.relax.op.op_attrs.ArgmaxArgminAttrs property)
- kernel_h (tvm.topi.nn.Workload attribute)
- kernel_layout (tvm.relax.op.op_attrs.Conv1DAttrs property)
- kernel_w (tvm.topi.nn.Workload attribute)
- keys() (tvm.ir.DictAttrs method)
- kill_storage() (in module tvm.relax.op.memory)
- kill_tensor() (in module tvm.relax.op.memory)
- KillAfterLastUse() (in module tvm.relax.transform)
- kind (tvm.tir.schedule.Instruction attribute)
L
- LambdaLift() (in module tvm.relax.transform)
- largest (tvm.relax.op.op_attrs.TopKAttrs property)
- layer_norm() (in module tvm.relax.op.nn)
- LayerNormAttrs (class in tvm.relax.op.op_attrs)
- Layout (class in tvm.tir)
- layout (tvm.relax.op.op_attrs.AdaptivePool1DAttrs property)
- layout() (in module tvm.tir)
- layout_transform() (in module tvm.relax.op)
- LayoutTransformAttrs (class in tvm.relax.op.op_attrs)
- LazyGetInput() (in module tvm.relax.transform)
- LazySetOutput() (in module tvm.relax.transform)
- LazyTransformParams (class in tvm.relax.transform)
- ldexp() (in module tvm.tir)
- LE (class in tvm.tir)
- leaky_relu() (in module tvm.topi.nn)
- leakyrelu() (in module tvm.relax.op.nn)
- 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.tir)
- LetExpr (tvm.tir.transform.HoistedLetBindings attribute)
- LetStmt (class in tvm.tir)
- LiftThreadBinding() (in module tvm.tir.transform)
- LiftTransformParams() (in module tvm.relax.transform)
- likely() (in module tvm.tir)
- linear() (in module tvm.relax.op)
- list_configs() (tvm.transform.PassContext static method)
- list_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)
- listdir() (tvm.contrib.utils.TempDirectory method)
- load() (tvm.meta_schedule.CostModel method)
- load_json() (in module tvm.ir)
- load_module() (tvm.rpc.RPCSession method)
- load_vm_module() (tvm.runtime.disco.Session method)
- LocalSession (class in tvm.rpc)
- log() (in module tvm.relax.op)
- log10() (in module tvm.te)
- log1p() (in module tvm.tir)
- log2() (in module tvm.te)
- log_add_exp() (in module tvm.relax.op)
- log_softmax() (in module tvm.contrib.miopen)
- logaddexp() (in module tvm.te)
- 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)
- lookup_param() (in module tvm.tir)
- loop_partition() (tvm.tir.schedule.Schedule method)
- LoopPartition() (in module tvm.tir.transform)
- LoopPartitionConfig (class in tvm.tir.transform)
- LoopRV (class in tvm.tir.schedule)
- LowerAllocTensor() (in module tvm.relax.transform)
- LowerAsyncDMA() (in module tvm.tir.transform)
- LowerAutoCopy() (in module tvm.tir.transform)
- LowerCrossThreadReduction() (in module tvm.tir.transform)
- LowerCustomDatatypes() (in module tvm.tir.transform)
- LowerDeviceKernelLaunch() (in module tvm.tir.transform)
- LowerDeviceStorageAccessInfo() (in module tvm.tir.transform)
- LowerGPUIPCAllocStorage (class in tvm.relax.transform)
- LowerInitBlock() (in module tvm.tir.transform)
- LowerIntrin() (in module tvm.tir.transform)
- LowerMatchBuffer() (in module tvm.tir.transform)
- LowerOpaqueBlock() (in module tvm.tir.transform)
- LowerRuntimeBuiltin() (in module tvm.relax.transform)
- LowerThreadAllreduce() (in module tvm.tir.transform)
- LowerTVMBuiltin() (in module tvm.tir.transform)
- LowerVtcmAlloc() (in module tvm.tir.transform)
- LowerWarpMemory() (in module tvm.tir.transform)
- lrn() (in module tvm.topi.nn)
- lstm() (in module tvm.topi.nn)
- LT (class in tvm.tir)
M
- make_closure() (in module tvm.relax.op)
- make_compilation_config() (in module tvm.target)
- make_filled_simdgroup_matrix() (in module tvm.tir)
- make_node() (in module tvm.ir)
- MakePackedAPI() (in module tvm.tir.transform)
- MakeUnpackedAPI() (in module tvm.tir.transform)
- mali() (in module tvm.target)
- ManifestSharedMemoryLocalStage() (in module tvm.tir.transform)
- Map (class in tvm.ir)
- map_indices() (tvm.tir.IndexMap method)
- map_shape() (tvm.tir.IndexMap method)
- map_tensor() (tvm.tir.IndexMap method)
- masked_fill() (in module tvm.relax.op)
- match_cast() (tvm.relax.block_builder.BlockBuilder method)
- MatchBufferRegion (class in tvm.tir)
- MatchCast (class in tvm.relax)
- matmul() (in module tvm.contrib.cblas)
- MatmulAttrs (class in tvm.relax.op.op_attrs)
- mattr (tvm.target.Target property)
- Max (class in tvm.tir)
- max() (in module tvm.relax.op)
- max_block_size_x (tvm.target.Target property)
- max_block_size_y (tvm.target.Target property)
- max_num_threads (tvm.target.Target property)
- max_pool1d() (in module tvm.relax.op.nn)
- max_pool2d() (in module tvm.relax.op.nn)
- max_pool2d_backward() (in module tvm.relax.op.grad)
- max_pool3d() (in module tvm.relax.op.nn)
- max_simplification_steps (tvm.tir.transform.RemoveNoOpConfig property)
- max_value() (in module tvm.te)
- maximum() (in module tvm.relax.op)
- mcpu (tvm.target.Target property)
- mean() (in module tvm.relax.op)
- MeasureCallback (class in tvm.meta_schedule)
- MeasureCandidate (class in tvm.meta_schedule)
- memoize() (in module tvm.contrib.pickle_memoize)
- memory_scope (tvm.target.VirtualDevice property)
- merge() (tvm.tir.schedule.Schedule method)
- MergeCompositeFunctions() (in module tvm.relax.transform)
- MergeSharedMemoryAllocations() (in module tvm.tir.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)
- MetricCollector (class in tvm.runtime.profiling)
- Min (class in tvm.tir)
- min() (in module tvm.relax.op)
- min_value() (in module tvm.te)
- minimum() (in module tvm.relax.op)
- mirror_pad() (in module tvm.topi.nn)
- mma_fill() (in module tvm.tir)
- mma_store() (in module tvm.tir)
- Mod (class in tvm.tir)
- mod (tvm.tir.schedule.Schedule property)
- mod() (in module tvm.relax.op)
- mode (tvm.relax.op.op_attrs.TakeAttrs property)
- model (tvm.target.Target property)
- module
- tvm.contrib
- tvm.contrib.cblas
- tvm.contrib.cc
- tvm.contrib.clang
- tvm.contrib.cublas
- tvm.contrib.dlpack
- tvm.contrib.emcc
- tvm.contrib.miopen
- tvm.contrib.ndk
- tvm.contrib.nnpack
- tvm.contrib.nvcc
- tvm.contrib.pickle_memoize
- tvm.contrib.random
- tvm.contrib.rocblas
- tvm.contrib.rocm
- tvm.contrib.spirv
- tvm.contrib.tar
- tvm.contrib.utils
- tvm.contrib.xcode
- tvm.dlight
- tvm.driver
- tvm.error
- tvm.instrument
- tvm.ir
- tvm.meta_schedule
- tvm.relax
- tvm.relax.analysis
- tvm.relax.block_builder
- tvm.relax.frontend
- tvm.relax.frontend.onnx
- tvm.relax.frontend.stablehlo
- tvm.relax.frontend.torch
- tvm.relax.op
- tvm.relax.op.builtin
- tvm.relax.op.ccl
- tvm.relax.op.distributed
- tvm.relax.op.grad
- tvm.relax.op.image
- tvm.relax.op.memory
- tvm.relax.op.nn
- tvm.relax.op.op_attrs
- tvm.relax.transform
- tvm.rpc
- tvm.runtime
- tvm.runtime.disco
- tvm.runtime.profiling
- tvm.runtime.vm
- tvm.target
- tvm.te
- tvm.tir
- tvm.tir.analysis.analysis
- tvm.tir.schedule
- tvm.tir.stmt_functor
- tvm.tir.transform
- tvm.topi.image
- tvm.topi.nn
- tvm.transform
- module_pass() (in module tvm.transform)
- ModulePass (class in tvm.transform)
- momentum (tvm.relax.op.op_attrs.BatchNormAttrs property)
- Mul (class in tvm.tir)
- multinomial_from_uniform() (in module tvm.relax.op)
- MultinomialFromUniformAttrs (class in tvm.relax.op.op_attrs)
- multiply() (in module tvm.relax.op)
- mutate_irmodule() (tvm.relax.DataflowBlockRewrite method)
- mutated_dfb() (tvm.relax.DataflowBlockRewrite method)
- mutated_root_fn() (tvm.relax.DataflowBlockRewrite method)
- Mutator (class in tvm.meta_schedule)
N
- name (tvm.tir.schedule.InstructionKind attribute)
- name_hint (tvm.relax.Var property)
- name_to_binding() (in module tvm.relax.analysis)
- namedtuple() (in module tvm.topi.nn)
- NarrowDataType() (in module tvm.tir.transform)
- nchw_pack_layout() (in module tvm.topi.image)
- nchw_xc_layout() (in module tvm.topi.image)
- ndim (tvm.te.Tensor property)
- NE (class in tvm.tir)
- nearbyint() (in module tvm.te)
- negative() (in module tvm.relax.op)
- Never (tvm.tir.transform.HoistedConditionals attribute)
- next_task_id() (tvm.meta_schedule.TaskScheduler method)
- nextafter() (in module tvm.tir)
- nll_loss() (in module tvm.relax.op.nn)
- 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)
- no_unroll_loop_with_extent_one (tvm.tir.transform.LoopPartitionConfig property)
- Node (class in tvm.ir)
- non_surjective_inverse() (tvm.tir.IndexMap method)
- nonzero() (in module tvm.relax.op)
- normal() (in module tvm.contrib.random)
- Normalize() (in module tvm.relax.transform)
- normalize() (tvm.relax.block_builder.BlockBuilder method)
- normalize_file_list_by_unpacking_tars() (in module tvm.contrib.tar)
- normalize_prim_func() (in module tvm.dlight)
- NormalizeGlobalVar() (in module tvm.relax.transform)
- Not (class in tvm.tir)
- not_equal() (in module tvm.relax.op)
- notify_runner_results() (tvm.meta_schedule.SearchStrategy method)
- 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)
O
- ObjectStructInfo (class in tvm.relax)
- ObjectType (class in tvm.relax)
- offset_of() (tvm.tir.Buffer method)
- one_hot() (in module tvm.relax.op)
- 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.tir.analysis.analysis)
- Op (class in tvm.ir)
- op_type (tvm.relax.op.op_attrs.AllReduceAttrs property)
- OPAQUE (tvm.tir.schedule.DepKind attribute)
- opaque_func() (tvm.relax.FuncStructInfo static method)
- OpAttributeInvalid
- OpAttributeRequired
- OpAttributeUnImplemented
- OpError
- OpNotImplemented
- optimize() (in module tvm.contrib.spirv)
- OptimizeLayoutTransform (class in tvm.relax.transform)
- options (tvm.target.TargetKind property)
- options_from_name() (tvm.target.TargetKind static method)
- Or (class in tvm.tir)
- out_dtype (tvm.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.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.Pool1DAttrs property)
- (tvm.relax.op.op_attrs.Pool2DAttrs property)
- (tvm.relax.op.op_attrs.Pool3DAttrs property)
- outer() (in module tvm.relax.op)
- output_padding (tvm.relax.op.op_attrs.Conv1DTransposeAttrs property)
- output_size (tvm.relax.op.op_attrs.AdaptivePool1DAttrs property)
- outputs (tvm.tir.schedule.Instruction attribute)
- override_instruments() (tvm.transform.PassContext method)
P
- PackedFuncType (class in tvm.relax)
- pad() (in module tvm.relax.op.nn)
- pad_einsum() (tvm.tir.schedule.Schedule method)
- pad_mode (tvm.relax.op.op_attrs.PadAttrs property)
- pad_value (tvm.relax.op.op_attrs.LayoutTransformAttrs 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.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.tir.schedule.Schedule method)
- parent (tvm.tir.schedule.StmtSRef property)
- parse_compute_version() (in module tvm.contrib.nvcc)
- partition_const_loop (tvm.tir.transform.LoopPartitionConfig property)
- Pass (class in tvm.transform)
- pass_instrument() (in module tvm.instrument)
- PassContext (class in tvm.transform)
- PassInfo (class in tvm.transform)
- PassInstrument (class in tvm.instrument)
- PassPrintingInstrument (class in tvm.instrument)
- PassTimingInstrument (class in tvm.instrument)
- PatternCheckContext (class in tvm.relax.transform)
- Percent (class in tvm.runtime.profiling)
- permute_dims() (in module tvm.relax.op)
- PermuteDimsAttrs (class in tvm.relax.op.op_attrs)
- 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)
- PlanAndUpdateBufferAllocationLocation() (in module tvm.tir.transform)
- PointerType (class in tvm.ir)
- PointerValueTypeRewrite() (in module tvm.tir.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_size (tvm.relax.op.op_attrs.Pool1DAttrs property)
- pop() (tvm.tir.schedule.Trace method)
- popcount() (in module tvm.te)
- PopenSession (class in tvm.rpc)
- post_order_visit() (in module tvm.relax.analysis)
- post_tuning() (tvm.meta_schedule.SearchStrategy method)
- Postproc (class in tvm.meta_schedule)
- pow() (in module tvm.tir)
- power() (in module tvm.relax.op)
- pre_order_visit() (in module tvm.tir.stmt_functor)
- pre_tuning() (tvm.meta_schedule.SearchStrategy method)
- predict() (tvm.meta_schedule.CostModel method)
- prelu() (in module tvm.relax.op.nn)
- PReluAttrs (class in tvm.relax.op.op_attrs)
- prim_func_pass() (in module tvm.tir.transform)
- PrimExpr (class in tvm.ir)
- PrimFunc (class in tvm.tir)
- PrimFuncPass (class in tvm.tir.transform)
- PrimStructInfo (class in tvm.relax)
- PrimType (class in tvm.ir)
- PrimValue (class in tvm.relax)
- print() (in module tvm.relax.op)
- print_tuning_statistics() (tvm.meta_schedule.TaskScheduler method)
- PrintAfterAll (class in tvm.instrument)
- PrintBeforeAll (class in tvm.instrument)
- PrintIR() (in module tvm.transform)
- ProcessSession (class in tvm.runtime.disco)
- prod() (in module tvm.relax.op)
- ProducerLoad (class in tvm.tir)
- profile() (tvm.relax.VirtualMachine method)
- profile_function() (in module tvm.runtime.profiling)
- Profiler (class in tvm.meta_schedule)
- propagate_knowns_to_prove_conditional (tvm.tir.transform.SimplifyConfig property)
- propagate_knowns_to_simplify_expressions (tvm.tir.transform.SimplifyConfig property)
- ptx_arrive_barrier() (in module tvm.tir)
- ptx_arrive_barrier_expect_tx() (in module tvm.tir)
- ptx_commit_group() (in module tvm.tir)
- ptx_cp_async() (in module tvm.tir)
- ptx_cp_async_barrier() (in module tvm.tir)
- ptx_cp_async_bulk() (in module tvm.tir)
- ptx_init_barrier_thread_count() (in module tvm.tir)
- ptx_ldmatrix() (in module tvm.tir)
- ptx_mma() (in module tvm.tir)
- ptx_mma_sp() (in module tvm.tir)
- ptx_wait_barrier() (in module tvm.tir)
- ptx_wait_group() (in module tvm.tir)
- PyExprMutator (class in tvm.relax)
- PyExprVisitor (class in tvm.relax)
- PyStmtExprMutator (class in tvm.tir)
- PyStmtExprVisitor (class in tvm.tir)
Q
- q_multiply_shift() (in module tvm.tir)
- q_multiply_shift_per_axis() (in module tvm.tir)
- quantize() (in module tvm.relax.op)
- QuantizeAttrs (class in tvm.relax.op.op_attrs)
- query() (tvm.meta_schedule.Database method)
- query_ir_module() (tvm.meta_schedule.Database method)
- query_schedule() (tvm.meta_schedule.Database method)
- query_tuning_record() (tvm.meta_schedule.Database method)
R
- r() (tvm.relax.ExecBuilder method)
- Ramp (class in tvm.tir)
- randint() (in module tvm.contrib.random)
- Range (class in tvm.ir)
- rasp() (in module tvm.target)
- rate (tvm.relax.op.op_attrs.DropoutAttrs property)
- Ratio (class in tvm.runtime.profiling)
- RAW (tvm.tir.schedule.DepKind attribute)
- RealizeVDevice() (in module tvm.relax.transform)
- redistribute() (in module tvm.relax.op.distributed)
- redistribute_replica_to_shard() (in module tvm.relax.op.distributed)
- Reduce (class in tvm.tir)
- reduce() (in module tvm.topi.nn)
- reduce_axis() (in module tvm.te)
- ReduceBranchingThroughOvercompute() (in module tvm.tir.transform)
- ReduceBranchingThroughOvercomputeConfig (class in tvm.tir.transform)
- reduction (tvm.relax.op.op_attrs.NLLLossAttrs property)
- reflect_pad() (in module tvm.topi.nn)
- register() (tvm.tir.TensorIntrin static method)
- register_error() (in module tvm.error)
- register_gradient() (in module tvm.relax.op)
- register_intrin_lowering() (in module tvm.ir)
- register_op_attr() (in module tvm.ir)
- register_pipeline() (in module tvm.relax)
- reindex() (tvm.tir.schedule.Schedule method)
- reindex_cache_read() (tvm.tir.schedule.Schedule method)
- reindex_cache_write() (tvm.tir.schedule.Schedule method)
- reinterpret() (in module tvm.tir)
- relax_dynamo() (in module tvm.relax.frontend.torch)
- RelaxExpr (class in tvm.ir)
- release() (tvm.contrib.utils.FileLock method)
- relpath() (tvm.contrib.utils.TempDirectory method)
- relu() (in module tvm.relax.op.nn)
- relu6() (in module tvm.relax.op.nn)
- remove() (tvm.contrib.utils.TempDirectory method)
- remove_all_unused() (in module tvm.relax.analysis)
- remove_rv() (tvm.tir.schedule.Schedule method)
- remove_unused() (tvm.relax.DataflowBlockRewrite method)
- RemoveAssume() (in module tvm.tir.transform)
- RemoveNoOp() (in module tvm.tir.transform)
- RemoveNoOpConfig (class in tvm.tir.transform)
- RemovePurityChecking() (in module tvm.relax.transform)
- RemoveRedundantReshape (class in tvm.relax.transform)
- RemoveStoreUndef() (in module tvm.tir.transform)
- RemoveUnusedOutputs() (in module tvm.relax.transform)
- RemoveUnusedParameters() (in module tvm.relax.transform)
- RemoveWeightLayoutRewriteBlock() (in module tvm.tir.transform)
- render() (tvm.instrument.PassTimingInstrument static method)
- renew_defs() (in module tvm.tir.stmt_functor)
- RenormalizeSplitPattern() (in module tvm.tir.transform)
- reorder() (tvm.tir.schedule.Schedule method)
- reorder_block_iter_var() (tvm.tir.schedule.Schedule method)
- ReorderPermuteDimsAfterConcat() (in module tvm.relax.transform)
- ReorderTakeAfterMatmul() (in module tvm.relax.transform)
- repeat() (in module tvm.relax.op)
- RepeatAttrs (class in tvm.relax.op.op_attrs)
- repeats (tvm.relax.op.op_attrs.RepeatAttrs property)
- replace() (tvm.tir.schedule.ScheduleState method)
- replace_all_uses() (tvm.relax.DataflowBlockRewrite method)
- replace_global_vars() (tvm.ir.IRModule method)
- replicate_pad() (in module tvm.topi.nn)
- Report (class in tvm.runtime.profiling)
- request() (tvm.rpc.TrackerSession method)
- request_and_run() (tvm.rpc.TrackerSession method)
- RequiredByConditional (tvm.tir.transform.HoistedLetBindings attribute)
- reset_attr() (tvm.ir.Op method)
- reshape() (in module tvm.relax.op)
- resize1d() (in module tvm.topi.image)
- resize2d() (in module tvm.relax.op.image)
- Resize2DAttrs (class in tvm.relax.op.op_attrs)
- resize3d() (in module tvm.topi.image)
- ret() (in module tvm.tir)
- ret_type (tvm.relax.op.op_attrs.TopKAttrs property)
- reverse_compute_at() (tvm.tir.schedule.Schedule method)
- reverse_compute_inline() (tvm.tir.schedule.Schedule method)
- RewriteCUDAGraph() (in module tvm.relax.transform)
- RewriteDataflowReshape() (in module tvm.relax.transform)
- RewriteUnsafeSelect() (in module tvm.tir.transform)
- rfactor() (tvm.tir.schedule.Schedule method)
- right_shift() (in module tvm.relax.op)
- riscv_cpu() (in module tvm.target)
- rms_norm() (in module tvm.relax.op.nn)
- RMSNormAttrs (class in tvm.relax.op.op_attrs)
- rocm() (in module tvm.target)
- rocm_link() (in module tvm.contrib.rocm)
- roi (tvm.relax.op.op_attrs.Resize2DAttrs property)
- rolling_buffer() (tvm.tir.schedule.Schedule method)
- root_mark() (tvm.tir.schedule.StmtSRef static method)
- round() (in module tvm.relax.op)
- rounding_method (tvm.relax.op.op_attrs.Resize2DAttrs property)
- RPCError
- RPCSession (class in tvm.rpc)
- RPCSessionTimeoutError
- rsqrt() (in module tvm.relax.op)
- run() (tvm.meta_schedule.Runner method)
- run_after_pass() (tvm.instrument.PassInstrument method)
- run_before_pass() (tvm.instrument.PassInstrument method)
- RunCodegen() (in module tvm.relax.transform)
- Runner (class in tvm.meta_schedule)
S
- same_as() (tvm.ir.GlobalInfo method)
- sample_categorical() (tvm.tir.schedule.Schedule method)
- sample_compute_location() (tvm.tir.schedule.Schedule method)
- sample_partitioned_tile() (tvm.tir.schedule.Schedule method)
- sample_perfect_tile() (tvm.tir.schedule.Schedule method)
- save() (tvm.meta_schedule.CostModel method)
- save_function() (tvm.relax.VirtualMachine method)
- save_json() (in module tvm.ir)
- scale (tvm.relax.op.op_attrs.AttentionAttrs 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)
- scan_axis (tvm.te.ScanOp property)
- ScanOp (class in tvm.te)
- ScanopAttrs (class in tvm.relax.op.op_attrs)
- scatter() (tvm.runtime.disco.Session method)
- scatter_elements() (in module tvm.relax.op)
- scatter_from_worker0() (in module tvm.relax.op.ccl)
- 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.tir.schedule)
- ScheduleDebugMask (class in tvm.tir.schedule)
- ScheduleError
- ScheduleRule (class in tvm.dlight)
- ScheduleState (class in tvm.tir.schedule)
- scope() (tvm.tir.Buffer method)
- script() (tvm.relax.BasePyModule method)
- SearchStrategy (class in tvm.meta_schedule)
- seed() (tvm.tir.schedule.Schedule method)
- Select (class in tvm.tir)
- selu() (in module tvm.relax.op.nn)
- SeqExpr (class in tvm.relax)
- SeqStmt (class in tvm.tir)
- Sequential (class in tvm.transform)
- SequentialSpan (class in tvm.ir)
- Server (class in tvm.rpc)
- Session (class in tvm.runtime.disco)
- set_attr() (tvm.ir.Op method)
- set_attrs_type_key() (tvm.ir.Op method)
- set_axis_separator() (tvm.tir.schedule.Schedule method)
- set_input() (tvm.relax.VirtualMachine method)
- set_instrument() (tvm.relax.VirtualMachine method)
- set_keep_for_debug() (tvm.contrib.utils.TempDirectory class method)
- set_num_inputs() (tvm.ir.Op method)
- set_scope() (tvm.tir.schedule.Schedule method)
- set_support_level() (tvm.ir.Op method)
- set_var_remap() (tvm.relax.PyExprMutator method)
- shape_of() (in module tvm.relax.op)
- shape_to_tensor() (in module tvm.relax.op)
- ShapeExpr (class in tvm.relax)
- ShapeStructInfo (class in tvm.relax)
- ShapeType (class in tvm.relax)
- shift_left() (in module tvm.tir)
- shift_right() (in module tvm.tir)
- should_run() (tvm.instrument.PassInstrument method)
- show() (tvm.relax.BasePyModule method)
- Shuffle (class in tvm.tir)
- shutdown() (tvm.runtime.disco.Session method)
- sigmoid() (in module tvm.relax.op)
- sign() (in module tvm.relax.op)
- silu() (in module tvm.relax.op.nn)
- simdgroup_load() (in module tvm.tir)
- simdgroup_multiply_accumulate() (in module tvm.tir)
- simdgroup_store() (in module tvm.tir)
- simplified() (tvm.tir.schedule.Trace method)
- Simplify() (in module tvm.tir.transform)
- simplify() (in module tvm.topi.image)
- SimplifyConfig (class in tvm.tir.transform)
- simulated_dequantize() (in module tvm.topi.nn)
- simulated_quantize() (in module tvm.topi.nn)
- sin() (in module tvm.relax.op)
- sinh() (in module tvm.relax.op)
- size_var() (in module tvm.te)
- SizeVar (class in tvm.tir)
- SkipAssert() (in module tvm.tir.transform)
- slice_scatter() (in module tvm.relax.op)
- SliceScatterAttrs (class in tvm.relax.op.op_attrs)
- SocketSession (class in tvm.runtime.disco)
- softmax() (in module tvm.contrib.miopen)
- softmax_common() (in module tvm.topi.nn)
- SoftmaxAttrs (class in tvm.relax.op.op_attrs)
- softplus() (in module tvm.relax.op.nn)
- SoftplusAttrs (class in tvm.relax.op.op_attrs)
- sort() (in module tvm.relax.op)
- SortAttrs (class in tvm.relax.op.op_attrs)
- SourceName (class in tvm.ir)
- space_to_batch_nd() (in module tvm.topi.nn)
- space_to_depth() (in module tvm.topi.nn)
- SpaceGenerator (class in tvm.meta_schedule)
- Span (class in tvm.ir)
- specialize() (tvm.tir.PrimFunc method)
- split() (in module tvm.relax.op)
- split_loop (tvm.tir.transform.InjectDoubleBufferConfig property)
- SplitAttrs (class in tvm.relax.op.op_attrs)
- SplitCallTIRByPattern() (in module tvm.relax.transform)
- SplitHostDevice() (in module tvm.tir.transform)
- SplitLayoutRewritePreproc() (in module tvm.relax.transform)
- sqrt() (in module tvm.relax.op)
- 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.tir)
- state (tvm.tir.schedule.Schedule property)
- StaticPlanBlockMemory() (in module tvm.relax.transform)
- StatisticalAttrs (class in tvm.relax.op.op_attrs)
- stats() (tvm.relax.VMExecutable method)
- std() (in module tvm.relax.op)
- stm32() (in module tvm.target)
- Stmt (class in tvm.tir)
- stmt (tvm.tir.schedule.StmtSRef property)
- stmt_list() (in module tvm.tir)
- stmt_seq() (in module tvm.tir)
- StmtSRef (class in tvm.tir.schedule)
- stop_lift_params() (in module tvm.relax.op.builtin)
- storage_align() (tvm.tir.schedule.Schedule method)
- StorageRewrite() (in module tvm.tir.transform)
- stride_h (tvm.topi.nn.Workload attribute)
- stride_w (tvm.topi.nn.Workload attribute)
- strided_slice() (in module tvm.relax.op)
- 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.Pool1DAttrs property)
- (tvm.relax.op.op_attrs.Pool2DAttrs property)
- (tvm.relax.op.op_attrs.Pool3DAttrs property)
- StringImm (class in tvm.relax)
- struct_info (tvm.ir.RelaxExpr property)
- struct_info_base_check() (in module tvm.relax.analysis)
- struct_info_lca() (in module tvm.relax.analysis)
- StructInfo (class in tvm.relax)
- structural_equal() (in module tvm.ir)
- structural_hash() (in module tvm.ir)
- Sub (class in tvm.tir)
- subscripts (tvm.relax.op.op_attrs.EinsumAttrs property)
- substitute() (in module tvm.tir.stmt_functor)
- subtract() (in module tvm.relax.op)
- suggest_layout_transforms() (in module tvm.relax.analysis)
- sum() (in module tvm.relax.op)
- summary() (tvm.rpc.TrackerSession method)
- support_block_scope_hoisting (tvm.tir.transform.HoistIfThenElseConfig property)
- sync_worker_0() (tvm.runtime.disco.Session method)
- system_lib() (tvm.rpc.RPCSession method)
T
- table() (tvm.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)
- tanh() (in module tvm.relax.op)
- tar() (in module tvm.contrib.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.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.contrib.utils)
- TempDirectory (class in tvm.contrib.utils)
- Tensor (class in tvm.te)
- tensor_to_shape() (in module tvm.relax.op)
- TensorIntrin (class in tvm.tir)
- tensorize() (tvm.tir.schedule.Schedule method)
- TensorSlice (class in tvm.te)
- TensorStructInfo (class in tvm.relax)
- TensorType (class in tvm.relax)
- terminate() (tvm.rpc.Server method)
- terminate_task() (tvm.meta_schedule.TaskScheduler method)
- testing_scope() (tvm.relax.block_builder.BlockBuilder method)
- TestingScope (class in tvm.relax.block_builder)
- text_summary() (tvm.rpc.TrackerSession method)
- thread_axis() (in module tvm.te)
- thread_warp_size (tvm.target.Target property)
- ThreadedSession (class in tvm.runtime.disco)
- ThreadSync() (in module tvm.tir.transform)
- threshold (tvm.relax.op.op_attrs.SoftplusAttrs property)
- tile() (in module tvm.relax.op)
- TileAttrs (class in tvm.relax.op.op_attrs)
- time_evaluator() (tvm.relax.VirtualMachine method)
- timeit() (tvm.meta_schedule.Profiler static method)
- tir_vars_in_struct_info() (in module tvm.relax.analysis)
- to_pytorch_func() (in module tvm.contrib.dlpack)
- to_vdevice() (in module tvm.relax.op)
- ToMixedPrecision() (in module tvm.relax.transform)
- ToNonDataflow() (in module tvm.relax.transform)
- topk() (in module tvm.relax.op)
- TopKAttrs (class in tvm.relax.op.op_attrs)
- TopologicalSort() (in module tvm.relax.transform)
- touch_task() (tvm.meta_schedule.TaskScheduler method)
- ToVDeviceAttrs (class in tvm.relax.op.op_attrs)
- Trace (class in tvm.tir.schedule)
- trace (tvm.tir.schedule.Schedule property)
- trace() (in module tvm.te)
- TrackerSession (class in tvm.rpc)
- training (tvm.relax.op.op_attrs.BatchNormAttrs property)
- transform_block_layout() (tvm.tir.schedule.Schedule method)
- transform_layout() (tvm.tir.schedule.Schedule method)
- TransformMmaBufferLayout() (in module tvm.tir.transform)
- transitively_prove_inequalities (tvm.tir.transform.SimplifyConfig 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)
- truncdiv() (in module tvm.te)
- truncmod() (in module tvm.te)
- try_inline() (in module tvm.dlight)
- try_inline_contiguous_spatial() (in module tvm.dlight)
- tune() (tvm.meta_schedule.TaskScheduler method)
- tune_tasks() (in module tvm.meta_schedule)
- tune_tir() (in module tvm.meta_schedule)
- TuneContext (class in tvm.meta_schedule)
- Tuple (class in tvm.relax)
- TupleGetItem (class in tvm.relax)
- TupleStructInfo (class in tvm.relax)
- TupleType (class in tvm.ir)
- tvm.contrib
- tvm.contrib.cblas
- tvm.contrib.cc
- tvm.contrib.clang
- tvm.contrib.cublas
- tvm.contrib.dlpack
- tvm.contrib.emcc
- tvm.contrib.miopen
- tvm.contrib.ndk
- tvm.contrib.nnpack
- tvm.contrib.nvcc
- tvm.contrib.pickle_memoize
- tvm.contrib.random
- tvm.contrib.rocblas
- tvm.contrib.rocm
- tvm.contrib.spirv
- tvm.contrib.tar
- tvm.contrib.utils
- tvm.contrib.xcode
- tvm.dlight
- tvm.driver
- tvm.error
- tvm.instrument
- tvm.ir
- tvm.meta_schedule
- tvm.relax
- tvm.relax.analysis
- tvm.relax.block_builder
- tvm.relax.frontend
- tvm.relax.frontend.onnx
- tvm.relax.frontend.stablehlo
- tvm.relax.frontend.torch
- tvm.relax.op
- tvm.relax.op.builtin
- tvm.relax.op.ccl
- tvm.relax.op.distributed
- tvm.relax.op.grad
- tvm.relax.op.image
- tvm.relax.op.memory
- tvm.relax.op.nn
- tvm.relax.op.op_attrs
- tvm.relax.transform
- tvm.rpc
- tvm.runtime
- tvm.runtime.disco
- tvm.runtime.profiling
- tvm.runtime.vm
- tvm.target
- tvm.te
- tvm.tir
- tvm.tir.analysis.analysis
- tvm.tir.schedule
- tvm.tir.stmt_functor
- tvm.tir.transform
- tvm.topi.image
- tvm.topi.nn
- tvm.transform
- tvm_access_ptr() (in module tvm.tir)
- tvm_bmma_sync() (in module tvm.tir)
- tvm_fill_fragment() (in module tvm.tir)
- tvm_load_matrix_sync() (in module tvm.tir)
- tvm_mma_sync() (in module tvm.tir)
- tvm_stack_alloca() (in module tvm.tir)
- tvm_stack_make_array() (in module tvm.tir)
- tvm_stack_make_shape() (in module tvm.tir)
- tvm_store_matrix_sync() (in module tvm.tir)
- tvm_struct_get() (in module tvm.tir)
- tvm_struct_set() (in module tvm.tir)
- tvm_thread_allreduce() (in module tvm.tir)
- tvm_throw_last_error() (in module tvm.tir)
- tvm_tuple() (in module tvm.tir)
- TVMBackendAllocWorkspace() (in module tvm.tir)
- TVMBackendFreeWorkspace() (in module tvm.tir)
- TVMError
- Type (class in tvm.ir)
- type_annotation() (in module tvm.tir)
U
- udchain() (in module tvm.relax.analysis)
- unannotate() (tvm.tir.schedule.Schedule method)
- undef() (in module tvm.tir)
- undefined_vars() (in module tvm.tir.analysis.analysis)
- uniform() (in module tvm.contrib.random)
- UnifyThreadBinding() (in module tvm.tir.transform)
- unique() (in module tvm.relax.op)
- unpack_NCHWc_to_nchw() (in module tvm.topi.nn)
- unroll() (tvm.tir.schedule.Schedule method)
- unroll_local_access (tvm.tir.transform.UnrollLoopConfig property)
- unroll_loop_with_partition_hint_no_interval (tvm.tir.transform.LoopPartitionConfig property)
- UnrollLoop() (in module tvm.tir.transform)
- UnrollLoopConfig (class in tvm.tir.transform)
- unsafe_hide_buffer_access() (tvm.tir.schedule.Schedule method)
- unsafe_set_dtype() (tvm.tir.schedule.Schedule method)
- untar() (in module tvm.contrib.tar)
- update() (tvm.ir.IRModule method)
- update_func() (tvm.ir.IRModule method)
- update_global_info() (tvm.ir.IRModule method)
- UpdateParamStructInfo() (in module tvm.relax.transform)
- UpdateVDevice() (in module tvm.relax.transform)
- upload() (tvm.rpc.RPCSession method)
- upsampling() (in module tvm.topi.nn)
- upsampling3d() (in module tvm.topi.nn)
- upscale_factor (tvm.relax.op.op_attrs.PixelShuffleAttrs property)
- use_dataflow_analysis (tvm.tir.transform.ReduceBranchingThroughOvercomputeConfig property)
- UseAssumeToReduceBranches() (in module tvm.tir.transform)
- UsingBlockVar (tvm.tir.transform.HoistedConditionals attribute)
V
- values() (tvm.ir.Map method)
- Var (class in tvm.relax)
- var() (in module tvm.te)
- VarBinding (class in tvm.relax)
- variance() (in module tvm.relax.op)
- VDevice (class in tvm.ir)
- vectorcombine() (in module tvm.tir)
- vectorhigh() (in module tvm.tir)
- vectorize() (tvm.tir.schedule.Schedule method)
- VectorizeLoop() (in module tvm.tir.transform)
- vectorlow() (in module tvm.tir)
- VERIFY_CACHED_FLAGS (tvm.tir.schedule.ScheduleDebugMask attribute)
- verify_gpu_code() (in module tvm.tir.analysis.analysis)
- verify_memory() (in module tvm.tir.analysis.analysis)
- VERIFY_SREF_TREE (tvm.tir.schedule.ScheduleDebugMask attribute)
- verify_ssa() (in module tvm.tir.analysis.analysis)
- verify_well_formed() (in module tvm.tir.analysis.analysis)
- VerifyMemory() (in module tvm.tir.transform)
- VerifyVTCMLimit() (in module tvm.tir.transform)
- view() (in module tvm.relax.op.memory)
- virtual_device_id (tvm.target.VirtualDevice property)
- VirtualDevice (class in tvm.target)
- VirtualMachine (class in tvm.relax)
- visit_add_() (tvm.tir.PyStmtExprMutator method)
- visit_allocate_() (tvm.tir.PyStmtExprMutator method)
- visit_allocate_const_() (tvm.tir.PyStmtExprMutator method)
- visit_and_() (tvm.tir.PyStmtExprMutator method)
- visit_assert_stmt_() (tvm.tir.PyStmtExprMutator method)
- visit_attr_stmt_() (tvm.tir.PyStmtExprMutator method)
- visit_binding() (tvm.relax.PyExprMutator method)
- visit_binding_block() (tvm.relax.PyExprMutator method)
- visit_binding_block_() (tvm.relax.PyExprMutator method)
- visit_block_() (tvm.tir.PyStmtExprMutator method)
- visit_block_realize_() (tvm.tir.PyStmtExprMutator method)
- visit_broadcast_() (tvm.tir.PyStmtExprMutator method)
- visit_buffer_load_() (tvm.tir.PyStmtExprMutator method)
- visit_buffer_realize_() (tvm.tir.PyStmtExprMutator method)
- visit_buffer_store_() (tvm.tir.PyStmtExprMutator method)
- visit_call_() (tvm.relax.PyExprMutator method)
- visit_cast_() (tvm.tir.PyStmtExprMutator method)
- visit_constant_() (tvm.relax.PyExprMutator method)
- visit_data_type_imm_() (tvm.relax.PyExprMutator method)
- visit_dataflow_block_() (tvm.relax.PyExprMutator method)
- visit_dataflow_var_() (tvm.relax.PyExprMutator method)
- visit_dataflow_var_def_() (tvm.relax.PyExprMutator method)
- visit_decl_buffer_() (tvm.tir.PyStmtExprMutator method)
- visit_div_() (tvm.tir.PyStmtExprMutator method)
- visit_eq_() (tvm.tir.PyStmtExprMutator method)
- visit_evaluate_() (tvm.tir.PyStmtExprMutator method)
- visit_expr() (tvm.relax.ExprFunctor method)
- visit_expr_post_order() (tvm.relax.PyExprMutator method)
- visit_extern_func_() (tvm.relax.PyExprMutator method)
- visit_float_imm_() (tvm.tir.PyStmtExprMutator method)
- visit_floor_div_() (tvm.tir.PyStmtExprMutator method)
- visit_floor_mod_() (tvm.tir.PyStmtExprMutator method)
- visit_for_() (tvm.tir.PyStmtExprMutator method)
- visit_function_() (tvm.relax.PyExprMutator method)
- visit_ge_() (tvm.tir.PyStmtExprMutator method)
- visit_global_var_() (tvm.relax.PyExprMutator method)
- visit_gt_() (tvm.tir.PyStmtExprMutator method)
- visit_if_() (tvm.relax.PyExprMutator method)
- visit_if_then_else_() (tvm.tir.PyStmtExprMutator method)
- visit_int_imm_() (tvm.tir.PyStmtExprMutator method)
- visit_le_() (tvm.tir.PyStmtExprMutator method)
- visit_let_() (tvm.tir.PyStmtExprMutator method)
- visit_let_stmt_() (tvm.tir.PyStmtExprMutator method)
- visit_lt_() (tvm.tir.PyStmtExprMutator method)
- visit_match_cast_() (tvm.relax.PyExprMutator method)
- visit_max_() (tvm.tir.PyStmtExprMutator method)
- visit_min_() (tvm.tir.PyStmtExprMutator method)
- visit_mod_() (tvm.tir.PyStmtExprMutator method)
- visit_mul_() (tvm.tir.PyStmtExprMutator method)
- visit_ne_() (tvm.tir.PyStmtExprMutator method)
- visit_not_() (tvm.tir.PyStmtExprMutator method)
- visit_op_() (tvm.relax.PyExprMutator method)
- visit_or_() (tvm.tir.PyStmtExprMutator method)
- visit_prim_value_() (tvm.relax.PyExprMutator method)
- visit_producer_load_() (tvm.tir.PyStmtExprMutator method)
- visit_ramp_() (tvm.tir.PyStmtExprMutator method)
- visit_reduce_() (tvm.tir.PyStmtExprMutator method)
- visit_select_() (tvm.tir.PyStmtExprMutator method)
- visit_seq_expr_() (tvm.relax.PyExprMutator method)
- visit_seq_stmt_() (tvm.tir.PyStmtExprMutator method)
- visit_shape_expr_() (tvm.relax.PyExprMutator method)
- visit_shuffle_() (tvm.tir.PyStmtExprMutator method)
- visit_size_var_() (tvm.tir.PyStmtExprMutator method)
- visit_span() (tvm.relax.PyExprMutator method)
- visit_stmt() (tvm.tir.PyStmtExprMutator method)
- visit_string_imm_() (tvm.relax.PyExprMutator method)
- visit_sub_() (tvm.tir.PyStmtExprMutator method)
- visit_tuple_() (tvm.relax.PyExprMutator method)
- visit_tuple_getitem_() (tvm.relax.PyExprMutator method)
- visit_var_() (tvm.relax.PyExprMutator method)
- visit_var_binding_() (tvm.relax.PyExprMutator method)
- visit_var_def() (tvm.relax.PyExprMutator method)
- visit_var_def_() (tvm.relax.PyExprMutator method)
- visit_while_() (tvm.tir.PyStmtExprMutator method)
- visit_with_new_scope() (tvm.relax.PyExprMutator method)
- vload() (tvm.tir.Buffer method)
- VMBuiltinLower() (in module tvm.relax.transform)
- VMExecutable (class in tvm.relax)
- VMInstrumentReturnKind (class in tvm.relax)
- VMShapeLower() (in module tvm.relax.transform)
- vscale() (in module tvm.tir)
- vstore() (tvm.tir.Buffer method)
- vulkan() (tvm.rpc.RPCSession method)
W
- WAR (tvm.tir.schedule.DepKind attribute)
- WAW (tvm.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.contrib.utils)
- While (class in tvm.tir)
- width (tvm.topi.nn.Workload attribute)
- window_size (tvm.relax.op.op_attrs.AttentionAttrs property)
- winograd_transform_matrices() (in module tvm.topi.nn)
- with_attr() (tvm.ir.BaseFunc method)
- with_attrs() (tvm.ir.BaseFunc method)
- with_body() (tvm.tir.PrimFunc method)
- with_decision() (tvm.tir.schedule.Trace method)
- with_minrpc() (in module tvm.rpc)
- with_struct_info() (tvm.relax.PyExprMutator method)
- without_attr() (tvm.ir.BaseFunc method)
- work_on() (tvm.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)