Path Lines of Code python/gen_requirements.py 373 python/pyproject.toml 22 python/setup.py 195 python/tvm/__init__.py 46 python/tvm/_ffi/__init__.py 5 python/tvm/_ffi/_pyversion.py 4 python/tvm/_ffi/base.py 31 python/tvm/_ffi/libinfo.py 160 python/tvm/_ffi/registry.py 7 python/tvm/arith/__init__.py 21 python/tvm/arith/_ffi_api.py 2 python/tvm/arith/analyzer.py 91 python/tvm/arith/bound.py 3 python/tvm/arith/int_set.py 34 python/tvm/arith/int_solver.py 35 python/tvm/arith/iter_affine_map.py 85 python/tvm/arith/pattern.py 9 python/tvm/contrib/__init__.py 1 python/tvm/contrib/cblas.py 34 python/tvm/contrib/cc.py 206 python/tvm/contrib/clang.py 44 python/tvm/contrib/coreml_runtime.py 18 python/tvm/contrib/cublas.py 29 python/tvm/contrib/cublaslt.py 17 python/tvm/contrib/cudnn.py 494 python/tvm/contrib/cutlass/__init__.py 1 python/tvm/contrib/cutlass/_ffi_api.py 2 python/tvm/contrib/cutlass/attention_operation.py 29 python/tvm/contrib/cutlass/build.py 655 python/tvm/contrib/cutlass/conv2d_operation.py 309 python/tvm/contrib/cutlass/conv2d_profiler.py 16 python/tvm/contrib/cutlass/gemm_operation.py 254 python/tvm/contrib/cutlass/gemm_profiler.py 15 python/tvm/contrib/cutlass/gen_conv2d.py 315 python/tvm/contrib/cutlass/gen_gemm.py 272 python/tvm/contrib/cutlass/gen_tensor_op.py 724 python/tvm/contrib/cutlass/layer_norm_operation.py 4 python/tvm/contrib/cutlass/library.py 208 python/tvm/contrib/cutlass/rms_norm_operation.py 4 python/tvm/contrib/dlpack.py 14 python/tvm/contrib/dnnl.py 79 python/tvm/contrib/download.py 83 python/tvm/contrib/emcc.py 46 python/tvm/contrib/hexagon/__init__.py 1 python/tvm/contrib/hexagon/_ci_env_check.py 17 python/tvm/contrib/hexagon/build.py 504 python/tvm/contrib/hexagon/generate_take_op.py 65 python/tvm/contrib/hexagon/hexagon_profiler.py 65 python/tvm/contrib/hexagon/hexagon_unary_ops.py 61 python/tvm/contrib/hexagon/meta_schedule.py 121 python/tvm/contrib/hexagon/profiling/process_lwp_data.py 270 python/tvm/contrib/hexagon/pytest_plugin.py 248 python/tvm/contrib/hexagon/session.py 139 python/tvm/contrib/hexagon/tools.py 345 python/tvm/contrib/hipblas.py 29 python/tvm/contrib/miopen.py 87 python/tvm/contrib/mkl.py 46 python/tvm/contrib/mps.py 31 python/tvm/contrib/mrvl.py 325 python/tvm/contrib/msc/__init__.py 1 python/tvm/contrib/msc/core/__init__.py 1 python/tvm/contrib/msc/core/_ffi_api.py 2 python/tvm/contrib/msc/core/codegen/__init__.py 2 python/tvm/contrib/msc/core/codegen/codegen.py 117 python/tvm/contrib/msc/core/codegen/sources.py 7 python/tvm/contrib/msc/core/frontend/__init__.py 1 python/tvm/contrib/msc/core/frontend/translate.py 153 python/tvm/contrib/msc/core/gym/__init__.py 3 python/tvm/contrib/msc/core/gym/agent/__init__.py 2 python/tvm/contrib/msc/core/gym/agent/base_agent.py 111 python/tvm/contrib/msc/core/gym/agent/method.py 21 python/tvm/contrib/msc/core/gym/agent/search_agent.py 65 python/tvm/contrib/msc/core/gym/control/__init__.py 2 python/tvm/contrib/msc/core/gym/control/configer.py 34 python/tvm/contrib/msc/core/gym/control/controller.py 43 python/tvm/contrib/msc/core/gym/control/namespace.py 15 python/tvm/contrib/msc/core/gym/control/service.py 357 python/tvm/contrib/msc/core/gym/control/worker.py 98 python/tvm/contrib/msc/core/gym/environment/__init__.py 3 python/tvm/contrib/msc/core/gym/environment/base_env.py 159 python/tvm/contrib/msc/core/gym/environment/method.py 70 python/tvm/contrib/msc/core/gym/environment/prune_env.py 28 python/tvm/contrib/msc/core/gym/environment/quantize_env.py 25 python/tvm/contrib/msc/core/gym/namespace.py 15 python/tvm/contrib/msc/core/ir/__init__.py 1 python/tvm/contrib/msc/core/ir/graph.py 329 python/tvm/contrib/msc/core/runtime/__init__.py 2 python/tvm/contrib/msc/core/runtime/hook.py 52 python/tvm/contrib/msc/core/runtime/jit.py 129 python/tvm/contrib/msc/core/runtime/runner.py 614 python/tvm/contrib/msc/core/tools/__init__.py 6 python/tvm/contrib/msc/core/tools/configer.py 26 python/tvm/contrib/msc/core/tools/distill/__init__.py 3 python/tvm/contrib/msc/core/tools/distill/configer.py 22 python/tvm/contrib/msc/core/tools/distill/distiller.py 95 python/tvm/contrib/msc/core/tools/distill/method.py 28 python/tvm/contrib/msc/core/tools/execute.py 129 python/tvm/contrib/msc/core/tools/prune/__init__.py 3 python/tvm/contrib/msc/core/tools/prune/configer.py 44 python/tvm/contrib/msc/core/tools/prune/method.py 46 python/tvm/contrib/msc/core/tools/prune/pruner.py 307 python/tvm/contrib/msc/core/tools/quantize/__init__.py 3 python/tvm/contrib/msc/core/tools/quantize/configer.py 74 python/tvm/contrib/msc/core/tools/quantize/method.py 202 python/tvm/contrib/msc/core/tools/quantize/quantizer.py 107 python/tvm/contrib/msc/core/tools/tool.py 612 python/tvm/contrib/msc/core/tools/track/__init__.py 3 python/tvm/contrib/msc/core/tools/track/configer.py 31 python/tvm/contrib/msc/core/tools/track/method.py 48 python/tvm/contrib/msc/core/tools/track/tracker.py 85 python/tvm/contrib/msc/core/transform/__init__.py 2 python/tvm/contrib/msc/core/transform/pattern.py 317 python/tvm/contrib/msc/core/transform/transform.py 25 python/tvm/contrib/msc/core/utils/__init__.py 9 python/tvm/contrib/msc/core/utils/arguments.py 120 python/tvm/contrib/msc/core/utils/dataset.py 274 python/tvm/contrib/msc/core/utils/expr.py 82 python/tvm/contrib/msc/core/utils/file.py 206 python/tvm/contrib/msc/core/utils/info.py 210 python/tvm/contrib/msc/core/utils/log.py 76 python/tvm/contrib/msc/core/utils/message.py 91 python/tvm/contrib/msc/core/utils/namespace.py 42 python/tvm/contrib/msc/core/utils/register.py 130 python/tvm/contrib/msc/framework/__init__.py 1 python/tvm/contrib/msc/framework/tensorflow/__init__.py 5 python/tvm/contrib/msc/framework/tensorflow/_ffi_api.py 2 python/tvm/contrib/msc/framework/tensorflow/codegen/__init__.py 1 python/tvm/contrib/msc/framework/tensorflow/codegen/codegen.py 27 python/tvm/contrib/msc/framework/tensorflow/frontend/__init__.py 1 python/tvm/contrib/msc/framework/tensorflow/frontend/translate.py 14 python/tvm/contrib/msc/framework/tensorflow/runtime/__init__.py 1 python/tvm/contrib/msc/framework/tensorflow/runtime/runner.py 130 python/tvm/contrib/msc/framework/tensorflow/tools/__init__.py 4 python/tvm/contrib/msc/framework/tensorflow/tools/distill/__init__.py 1 python/tvm/contrib/msc/framework/tensorflow/tools/distill/distiller.py 16 python/tvm/contrib/msc/framework/tensorflow/tools/prune/__init__.py 1 python/tvm/contrib/msc/framework/tensorflow/tools/prune/pruner.py 16 python/tvm/contrib/msc/framework/tensorflow/tools/quantize/__init__.py 1 python/tvm/contrib/msc/framework/tensorflow/tools/quantize/quantizer.py 16 python/tvm/contrib/msc/framework/tensorflow/tools/track/__init__.py 1 python/tvm/contrib/msc/framework/tensorflow/tools/track/tracker.py 16 python/tvm/contrib/msc/framework/tensorrt/__init__.py 1 python/tvm/contrib/msc/framework/tensorrt/_ffi_api.py 2 python/tvm/contrib/msc/framework/tensorrt/codegen/__init__.py 1 python/tvm/contrib/msc/framework/tensorrt/codegen/codegen.py 125 python/tvm/contrib/msc/framework/tensorrt/codegen/sources.py 21 python/tvm/contrib/msc/framework/tensorrt/codegen/utils.py 21 python/tvm/contrib/msc/framework/tensorrt/frontend/__init__.py 1 python/tvm/contrib/msc/framework/tensorrt/frontend/translate.py 30 python/tvm/contrib/msc/framework/tensorrt/runtime/__init__.py 1 python/tvm/contrib/msc/framework/tensorrt/runtime/runner.py 73 python/tvm/contrib/msc/framework/tensorrt/tools/__init__.py 4 python/tvm/contrib/msc/framework/tensorrt/tools/distill/__init__.py 1 python/tvm/contrib/msc/framework/tensorrt/tools/distill/distiller.py 16 python/tvm/contrib/msc/framework/tensorrt/tools/prune/__init__.py 1 python/tvm/contrib/msc/framework/tensorrt/tools/prune/pruner.py 16 python/tvm/contrib/msc/framework/tensorrt/tools/quantize/__init__.py 2 python/tvm/contrib/msc/framework/tensorrt/tools/quantize/method.py 59 python/tvm/contrib/msc/framework/tensorrt/tools/quantize/quantizer.py 196 python/tvm/contrib/msc/framework/tensorrt/tools/track/__init__.py 1 python/tvm/contrib/msc/framework/tensorrt/tools/track/tracker.py 54 python/tvm/contrib/msc/framework/tensorrt/transform/__init__.py 2 python/tvm/contrib/msc/framework/tensorrt/transform/pattern.py 245 python/tvm/contrib/msc/framework/tensorrt/transform/transform.py 13 python/tvm/contrib/msc/framework/torch/__init__.py 1 python/tvm/contrib/msc/framework/torch/_ffi_api.py 2 python/tvm/contrib/msc/framework/torch/codegen/__init__.py 1 python/tvm/contrib/msc/framework/torch/codegen/codegen.py 34 python/tvm/contrib/msc/framework/torch/frontend/__init__.py 1 python/tvm/contrib/msc/framework/torch/frontend/translate.py 40 python/tvm/contrib/msc/framework/torch/runtime/__init__.py 2 python/tvm/contrib/msc/framework/torch/runtime/jit.py 66 python/tvm/contrib/msc/framework/torch/runtime/runner.py 167 python/tvm/contrib/msc/framework/torch/tools/__init__.py 4 python/tvm/contrib/msc/framework/torch/tools/distill/__init__.py 2 python/tvm/contrib/msc/framework/torch/tools/distill/distiller.py 61 python/tvm/contrib/msc/framework/torch/tools/distill/method.py 46 python/tvm/contrib/msc/framework/torch/tools/prune/__init__.py 1 python/tvm/contrib/msc/framework/torch/tools/prune/pruner.py 16 python/tvm/contrib/msc/framework/torch/tools/quantize/__init__.py 2 python/tvm/contrib/msc/framework/torch/tools/quantize/method.py 127 python/tvm/contrib/msc/framework/torch/tools/quantize/quantizer.py 16 python/tvm/contrib/msc/framework/torch/tools/track/__init__.py 1 python/tvm/contrib/msc/framework/torch/tools/track/tracker.py 16 python/tvm/contrib/msc/framework/tvm/__init__.py 1 python/tvm/contrib/msc/framework/tvm/_ffi_api.py 2 python/tvm/contrib/msc/framework/tvm/codegen/__init__.py 1 python/tvm/contrib/msc/framework/tvm/codegen/codegen.py 14 python/tvm/contrib/msc/framework/tvm/runtime/__init__.py 1 python/tvm/contrib/msc/framework/tvm/runtime/runner.py 157 python/tvm/contrib/msc/framework/tvm/tools/__init__.py 4 python/tvm/contrib/msc/framework/tvm/tools/distill/__init__.py 1 python/tvm/contrib/msc/framework/tvm/tools/distill/distiller.py 16 python/tvm/contrib/msc/framework/tvm/tools/prune/__init__.py 1 python/tvm/contrib/msc/framework/tvm/tools/prune/pruner.py 16 python/tvm/contrib/msc/framework/tvm/tools/quantize/__init__.py 2 python/tvm/contrib/msc/framework/tvm/tools/quantize/method.py 87 python/tvm/contrib/msc/framework/tvm/tools/quantize/quantizer.py 67 python/tvm/contrib/msc/framework/tvm/tools/track/__init__.py 1 python/tvm/contrib/msc/framework/tvm/tools/track/tracker.py 58 python/tvm/contrib/msc/pipeline/__init__.py 2 python/tvm/contrib/msc/pipeline/dynamic.py 193 python/tvm/contrib/msc/pipeline/manager.py 65 python/tvm/contrib/msc/pipeline/pipeline.py 403 python/tvm/contrib/msc/pipeline/utils.py 105 python/tvm/contrib/msc/pipeline/worker.py 427 python/tvm/contrib/msc/pipeline/wrapper.py 168 python/tvm/contrib/msc/plugin/__init__.py 1 python/tvm/contrib/msc/plugin/_ffi_api.py 2 python/tvm/contrib/msc/plugin/build.py 123 python/tvm/contrib/msc/plugin/codegen/__init__.py 1 python/tvm/contrib/msc/plugin/codegen/codegen.py 177 python/tvm/contrib/msc/plugin/codegen/sources.py 21 python/tvm/contrib/msc/plugin/op/__init__.py 1 python/tvm/contrib/msc/plugin/op/_ffi_api.py 2 python/tvm/contrib/msc/plugin/register.py 42 python/tvm/contrib/msc/plugin/utils.py 43 python/tvm/contrib/ndk.py 86 python/tvm/contrib/nnpack.py 111 python/tvm/contrib/nvcc.py 230 python/tvm/contrib/pickle_memoize.py 75 python/tvm/contrib/popen_pool.py 206 python/tvm/contrib/random.py 32 python/tvm/contrib/rocblas.py 26 python/tvm/contrib/rocm.py 139 python/tvm/contrib/rpc.py 7 python/tvm/contrib/spirv.py 20 python/tvm/contrib/tar.py 53 python/tvm/contrib/tflite_runtime.py 28 python/tvm/contrib/thrust.py 19 python/tvm/contrib/tvmjs.py 269 python/tvm/contrib/utils.py 125 python/tvm/contrib/xcode.py 87 python/tvm/dlight/__init__.py 13 python/tvm/dlight/analysis/__init__.py 14 python/tvm/dlight/analysis/common_analysis.py 182 python/tvm/dlight/analysis/gemv.py 99 python/tvm/dlight/base/__init__.py 10 python/tvm/dlight/base/common_schedules.py 44 python/tvm/dlight/base/schedule_rule.py 37 python/tvm/dlight/base/transform.py 54 python/tvm/dlight/base/utils.py 68 python/tvm/dlight/benchmark/__init__.py 7 python/tvm/dlight/benchmark/bench.py 177 python/tvm/dlight/benchmark/extract.py 171 python/tvm/dlight/benchmark/utils.py 70 python/tvm/dlight/cpu/__init__.py 1 python/tvm/dlight/cpu/base.py 5 python/tvm/dlight/cpu/gemv.py 87 python/tvm/dlight/gpu/__init__.py 8 python/tvm/dlight/gpu/base.py 5 python/tvm/dlight/gpu/fallback.py 55 python/tvm/dlight/gpu/gemv.py 563 python/tvm/dlight/gpu/general_reduction.py 110 python/tvm/dlight/gpu/low_batch_gemv.py 611 python/tvm/dlight/gpu/matmul.py 918 python/tvm/dlight/gpu/reduction.py 244 python/tvm/dlight/gpu/rmsnorm.py 90 python/tvm/dlight/gpu/transpose.py 96 python/tvm/driver/__init__.py 1 python/tvm/driver/_ffi_api.py 2 python/tvm/driver/build_module.py 39 python/tvm/error.py 21 python/tvm/exec/__init__.py 1 python/tvm/exec/disco_remote_socket_session.py 12 python/tvm/exec/disco_worker.py 62 python/tvm/exec/gpu_memory_bandwidth.py 204 python/tvm/exec/popen_worker.py 66 python/tvm/exec/query_rpc_tracker.py 20 python/tvm/exec/rpc_proxy.py 78 python/tvm/exec/rpc_server.py 71 python/tvm/exec/rpc_tracker.py 17 python/tvm/ffi/__init__.py 44 python/tvm/ffi/_ffi_api.py 2 python/tvm/ffi/container.py 102 python/tvm/ffi/convert.py 30 python/tvm/ffi/cython/base.pxi 159 python/tvm/ffi/cython/core.pyx 8 python/tvm/ffi/cython/device.pxi 117 python/tvm/ffi/cython/dtype.pxi 60 python/tvm/ffi/cython/error.pxi 84 python/tvm/ffi/cython/function.pxi 214 python/tvm/ffi/cython/ndarray.pxi 172 python/tvm/ffi/cython/object.pxi 123 python/tvm/ffi/cython/string.pxi 31 python/tvm/ffi/dtype.py 73 python/tvm/ffi/error.py 52 python/tvm/ffi/ndarray.py 57 python/tvm/ffi/registry.py 59 python/tvm/generic.py 1 python/tvm/ir/__init__.py 28 python/tvm/ir/_ffi_analysis_api.py 2 python/tvm/ir/_ffi_api.py 2 python/tvm/ir/_ffi_instrument_api.py 2 python/tvm/ir/_ffi_transform_api.py 2 python/tvm/ir/analysis.py 7 python/tvm/ir/attrs.py 39 python/tvm/ir/base.py 74 python/tvm/ir/container.py 2 python/tvm/ir/diagnostics/__init__.py 38 python/tvm/ir/diagnostics/_ffi_api.py 2 python/tvm/ir/expr.py 75 python/tvm/ir/function.py 30 python/tvm/ir/global_info.py 25 python/tvm/ir/instrument.py 75 python/tvm/ir/json_compact.py 94 python/tvm/ir/module.py 91 python/tvm/ir/op.py 45 python/tvm/ir/supply.py 31 python/tvm/ir/transform.py 149 python/tvm/ir/type.py 32 python/tvm/ir/type_relation.py 11 python/tvm/meta_schedule/__init__.py 38 python/tvm/meta_schedule/_ffi_api.py 2 python/tvm/meta_schedule/arg_info.py 38 python/tvm/meta_schedule/builder/__init__.py 2 python/tvm/meta_schedule/builder/builder.py 65 python/tvm/meta_schedule/builder/local_builder.py 136 python/tvm/meta_schedule/cost_model/__init__.py 3 python/tvm/meta_schedule/cost_model/cost_model.py 106 python/tvm/meta_schedule/cost_model/metric.py 8 python/tvm/meta_schedule/cost_model/mlp_model.py 593 python/tvm/meta_schedule/cost_model/random_model.py 46 python/tvm/meta_schedule/cost_model/xgb_model.py 506 python/tvm/meta_schedule/database/__init__.py 6 python/tvm/meta_schedule/database/database.py 226 python/tvm/meta_schedule/database/json_database.py 34 python/tvm/meta_schedule/database/memory_database.py 13 python/tvm/meta_schedule/database/ordered_union_database.py 10 python/tvm/meta_schedule/database/schedule_fn_database.py 17 python/tvm/meta_schedule/database/union_database.py 10 python/tvm/meta_schedule/extracted_task.py 28 python/tvm/meta_schedule/feature_extractor/__init__.py 3 python/tvm/meta_schedule/feature_extractor/feature_extractor.py 48 python/tvm/meta_schedule/feature_extractor/per_store_feature.py 24 python/tvm/meta_schedule/feature_extractor/random_feature_extractor.py 29 python/tvm/meta_schedule/logging.py 156 python/tvm/meta_schedule/measure_callback/__init__.py 4 python/tvm/meta_schedule/measure_callback/add_to_database.py 9 python/tvm/meta_schedule/measure_callback/measure_callback.py 59 python/tvm/meta_schedule/measure_callback/remove_build_artifact.py 9 python/tvm/meta_schedule/measure_callback/update_cost_model.py 9 python/tvm/meta_schedule/mutator/__init__.py 6 python/tvm/meta_schedule/mutator/mutate_compute_location.py 9 python/tvm/meta_schedule/mutator/mutate_parallel.py 10 python/tvm/meta_schedule/mutator/mutate_thread_binding.py 9 python/tvm/meta_schedule/mutator/mutate_tile_size.py 9 python/tvm/meta_schedule/mutator/mutate_unroll.py 9 python/tvm/meta_schedule/mutator/mutator.py 67 python/tvm/meta_schedule/post_optimization/__init__.py 4 python/tvm/meta_schedule/post_optimization/droplet.py 82 python/tvm/meta_schedule/post_optimization/post_opt.py 26 python/tvm/meta_schedule/post_optimization/space.py 177 python/tvm/meta_schedule/post_optimization/utils.py 37 python/tvm/meta_schedule/postproc/__init__.py 11 python/tvm/meta_schedule/postproc/disallow_async_strided_mem_copy.py 9 python/tvm/meta_schedule/postproc/disallow_dynamic_loop.py 9 python/tvm/meta_schedule/postproc/postproc.py 61 python/tvm/meta_schedule/postproc/rewrite_cooperative_fetch.py 9 python/tvm/meta_schedule/postproc/rewrite_layout.py 9 python/tvm/meta_schedule/postproc/rewrite_parallel_vectorize_unroll.py 9 python/tvm/meta_schedule/postproc/rewrite_reduction_block.py 9 python/tvm/meta_schedule/postproc/rewrite_tensorize.py 10 python/tvm/meta_schedule/postproc/rewrite_unbound_block.py 10 python/tvm/meta_schedule/postproc/verify_gpu_code.py 9 python/tvm/meta_schedule/postproc/verify_vtcm_limit.py 9 python/tvm/meta_schedule/profiler.py 34 python/tvm/meta_schedule/relax_integration.py 191 python/tvm/meta_schedule/runner/__init__.py 12 python/tvm/meta_schedule/runner/config.py 103 python/tvm/meta_schedule/runner/local_runner.py 206 python/tvm/meta_schedule/runner/rpc_runner.py 270 python/tvm/meta_schedule/runner/runner.py 92 python/tvm/meta_schedule/runner/utils.py 57 python/tvm/meta_schedule/schedule/__init__.py 1 python/tvm/meta_schedule/schedule/cpu/__init__.py 1 python/tvm/meta_schedule/schedule/cuda/__init__.py 1 python/tvm/meta_schedule/schedule/cuda/layout_transform.py 222 python/tvm/meta_schedule/schedule/generic/__init__.py 1 python/tvm/meta_schedule/schedule/x86/__init__.py 1 python/tvm/meta_schedule/schedule_rule/__init__.py 15 python/tvm/meta_schedule/schedule_rule/add_rfactor.py 16 python/tvm/meta_schedule/schedule_rule/apply_custom_rule.py 9 python/tvm/meta_schedule/schedule_rule/auto_bind.py 20 python/tvm/meta_schedule/schedule_rule/auto_inline.py 34 python/tvm/meta_schedule/schedule_rule/cross_thread_reduction.py 11 python/tvm/meta_schedule/schedule_rule/multi_level_tiling.py 101 python/tvm/meta_schedule/schedule_rule/parallel_vectorize_unroll.py 22 python/tvm/meta_schedule/schedule_rule/random_compute_location.py 9 python/tvm/meta_schedule/schedule_rule/schedule_rule.py 63 python/tvm/meta_schedule/search_strategy/__init__.py 4 python/tvm/meta_schedule/search_strategy/evolutionary_search.py 35 python/tvm/meta_schedule/search_strategy/replay_func.py 9 python/tvm/meta_schedule/search_strategy/replay_trace.py 11 python/tvm/meta_schedule/search_strategy/search_strategy.py 150 python/tvm/meta_schedule/space_generator/__init__.py 5 python/tvm/meta_schedule/space_generator/post_order_apply.py 26 python/tvm/meta_schedule/space_generator/schedule_fn.py 26 python/tvm/meta_schedule/space_generator/space_generator.py 143 python/tvm/meta_schedule/space_generator/space_generator_union.py 27 python/tvm/meta_schedule/task_scheduler/__init__.py 3 python/tvm/meta_schedule/task_scheduler/gradient_based.py 21 python/tvm/meta_schedule/task_scheduler/round_robin.py 12 python/tvm/meta_schedule/task_scheduler/task_scheduler.py 137 python/tvm/meta_schedule/testing/__init__.py 1 python/tvm/meta_schedule/testing/custom_builder_runner.py 25 python/tvm/meta_schedule/testing/distributed_measure_candidates.py 151 python/tvm/meta_schedule/testing/dummy_object.py 31 python/tvm/meta_schedule/testing/local_rpc.py 37 python/tvm/meta_schedule/testing/space_generation.py 108 python/tvm/meta_schedule/testing/te_workload.py 692 python/tvm/meta_schedule/testing/tune_te.py 123 python/tvm/meta_schedule/testing/tune_utils.py 49 python/tvm/meta_schedule/testing/validate_database.py 437 python/tvm/meta_schedule/tir_integration.py 137 python/tvm/meta_schedule/trace_apply.py 5 python/tvm/meta_schedule/tune.py 67 python/tvm/meta_schedule/tune_context.py 157 python/tvm/meta_schedule/utils.py 161 python/tvm/relax/__init__.py 76 python/tvm/relax/_ffi_api.py 2 python/tvm/relax/analysis/__init__.py 29 python/tvm/relax/analysis/_ffi_api.py 2 python/tvm/relax/analysis/analysis.py 87 python/tvm/relax/analysis/estimate_memory_usage.py 99 python/tvm/relax/backend/__init__.py 4 python/tvm/relax/backend/_ffi_api.py 2 python/tvm/relax/backend/adreno/__init__.py 7 python/tvm/relax/backend/adreno/clml.py 445 python/tvm/relax/backend/adreno/pipeline.py 41 python/tvm/relax/backend/contrib/__init__.py 1 python/tvm/relax/backend/contrib/nnapi.py 218 python/tvm/relax/backend/cpu_generic/__init__.py 6 python/tvm/relax/backend/cpu_generic/pipeline.py 42 python/tvm/relax/backend/cuda/__init__.py 7 python/tvm/relax/backend/cuda/cublas.py 172 python/tvm/relax/backend/cuda/cudnn.py 128 python/tvm/relax/backend/cuda/cutlass.py 419 python/tvm/relax/backend/cuda/flashinfer.py 283 python/tvm/relax/backend/cuda/pipeline.py 54 python/tvm/relax/backend/dispatch_sampling.py 59 python/tvm/relax/backend/dispatch_sort_scan.py 192 python/tvm/relax/backend/gpu_generic/__init__.py 9 python/tvm/relax/backend/gpu_generic/cumsum.py 125 python/tvm/relax/backend/gpu_generic/pipeline.py 53 python/tvm/relax/backend/gpu_generic/sampling.py 243 python/tvm/relax/backend/metal/__init__.py 1 python/tvm/relax/backend/metal/coreml.py 326 python/tvm/relax/backend/pattern_registry.py 38 python/tvm/relax/backend/patterns.py 336 python/tvm/relax/backend/rocm/__init__.py 6 python/tvm/relax/backend/rocm/hipblas.py 120 python/tvm/relax/backend/rocm/pipeline.py 53 python/tvm/relax/backend/utils.py 51 python/tvm/relax/binding_rewrite.py 35 python/tvm/relax/block_builder.py 223 python/tvm/relax/distributed/__init__.py 3 python/tvm/relax/distributed/_ffi_api.py 2 python/tvm/relax/distributed/global_info.py 22 python/tvm/relax/distributed/struct_info.py 46 python/tvm/relax/distributed/transform/__init__.py 6 python/tvm/relax/distributed/transform/_ffi_api.py 2 python/tvm/relax/distributed/transform/transform.py 10 python/tvm/relax/dpl/__init__.py 9 python/tvm/relax/dpl/_ffi.py 2 python/tvm/relax/dpl/context.py 21 python/tvm/relax/dpl/pattern.py 312 python/tvm/relax/dpl/rewrite.py 62 python/tvm/relax/exec_builder.py 84 python/tvm/relax/expr.py 519 python/tvm/relax/expr_functor.py 469 python/tvm/relax/frontend/__init__.py 2 python/tvm/relax/frontend/common.py 69 python/tvm/relax/frontend/nn/__init__.py 23 python/tvm/relax/frontend/nn/_tensor_op.py 62 python/tvm/relax/frontend/nn/core.py 359 python/tvm/relax/frontend/nn/exporter.py 261 python/tvm/relax/frontend/nn/extern.py 211 python/tvm/relax/frontend/nn/llm/__init__.py 4 python/tvm/relax/frontend/nn/llm/kv_cache.py 2487 python/tvm/relax/frontend/nn/llm/position_embedding.py 383 python/tvm/relax/frontend/nn/llm/tree_attn.py 1058 python/tvm/relax/frontend/nn/modules.py 507 python/tvm/relax/frontend/nn/op.py 780 python/tvm/relax/frontend/nn/spec.py 156 python/tvm/relax/frontend/nn/subroutine.py 113 python/tvm/relax/frontend/nn/torch.py 97 python/tvm/relax/frontend/nn/visitor.py 38 python/tvm/relax/frontend/onnx/__init__.py 1 python/tvm/relax/frontend/onnx/onnx_frontend.py 2434 python/tvm/relax/frontend/stablehlo/__init__.py 1 python/tvm/relax/frontend/stablehlo/stablehlo_translator.py 318 python/tvm/relax/frontend/torch/__init__.py 3 python/tvm/relax/frontend/torch/base_fx_graph_translator.py 1460 python/tvm/relax/frontend/torch/dynamo.py 112 python/tvm/relax/frontend/torch/exported_program_translator.py 536 python/tvm/relax/frontend/torch/fx_translator.py 830 python/tvm/relax/ir/instrument.py 19 python/tvm/relax/op/__init__.py 137 python/tvm/relax/op/_ffi_api.py 2 python/tvm/relax/op/_op_gradient.py 738 python/tvm/relax/op/base.py 255 python/tvm/relax/op/binary.py 52 python/tvm/relax/op/builtin/__init__.py 1 python/tvm/relax/op/builtin/_ffi_api.py 2 python/tvm/relax/op/builtin/builtin.py 25 python/tvm/relax/op/ccl/__init__.py 1 python/tvm/relax/op/ccl/_ffi_api.py 2 python/tvm/relax/op/ccl/ccl.py 15 python/tvm/relax/op/create.py 77 python/tvm/relax/op/datatype.py 8 python/tvm/relax/op/distributed/__init__.py 6 python/tvm/relax/op/distributed/_ffi_api.py 2 python/tvm/relax/op/distributed/distributed.py 28 python/tvm/relax/op/grad/__init__.py 1 python/tvm/relax/op/grad/_ffi_api.py 2 python/tvm/relax/op/grad/grad.py 70 python/tvm/relax/op/image/__init__.py 1 python/tvm/relax/op/image/_ffi_api.py 2 python/tvm/relax/op/image/image.py 43 python/tvm/relax/op/index.py 25 python/tvm/relax/op/linear_algebra.py 19 python/tvm/relax/op/manipulate.py 111 python/tvm/relax/op/mask.py 5 python/tvm/relax/op/memory/__init__.py 2 python/tvm/relax/op/memory/_ffi_api.py 2 python/tvm/relax/op/memory/memory.py 33 python/tvm/relax/op/memory/view.py 22 python/tvm/relax/op/nn/__init__.py 37 python/tvm/relax/op/nn/_ffi_api.py 2 python/tvm/relax/op/nn/nn.py 535 python/tvm/relax/op/op_attrs.py 72 python/tvm/relax/op/qdq.py 10 python/tvm/relax/op/sampling.py 16 python/tvm/relax/op/search.py 9 python/tvm/relax/op/set.py 50 python/tvm/relax/op/sorting.py 17 python/tvm/relax/op/statistical.py 50 python/tvm/relax/op/ternary.py 4 python/tvm/relax/op/unary.py 68 python/tvm/relax/op/vm/__init__.py 1 python/tvm/relax/op/vm/_ffi_api.py 2 python/tvm/relax/op/vm/vm.py 34 python/tvm/relax/pipeline.py 164 python/tvm/relax/struct_info.py 116 python/tvm/relax/testing/__init__.py 4 python/tvm/relax/testing/ast_printer.py 255 python/tvm/relax/testing/attention.py 111 python/tvm/relax/testing/lib_comparator.py 58 python/tvm/relax/testing/matmul.py 40 python/tvm/relax/testing/nn.py 163 python/tvm/relax/testing/runtime_builtin.py 9 python/tvm/relax/testing/transform.py 62 python/tvm/relax/testing/vm.py 46 python/tvm/relax/training/__init__.py 7 python/tvm/relax/training/_ffi_api.py 2 python/tvm/relax/training/loss.py 150 python/tvm/relax/training/optimizer.py 283 python/tvm/relax/training/setup_trainer.py 92 python/tvm/relax/training/trainer.py 183 python/tvm/relax/training/utils.py 35 python/tvm/relax/transform/__init__.py 80 python/tvm/relax/transform/_ffi_api.py 2 python/tvm/relax/transform/attach_external_modules.py 22 python/tvm/relax/transform/fast_math.py 27 python/tvm/relax/transform/fold_batch_norm_to_conv2d_for_inference.py 54 python/tvm/relax/transform/fuse_transpose_matmul.py 121 python/tvm/relax/transform/ipc_allreduce_rewrite.py 90 python/tvm/relax/transform/lazy_transform_params.py 260 python/tvm/relax/transform/legalize_ops/__init__.py 16 python/tvm/relax/transform/legalize_ops/binary.py 37 python/tvm/relax/transform/legalize_ops/ccl.py 95 python/tvm/relax/transform/legalize_ops/common.py 28 python/tvm/relax/transform/legalize_ops/create.py 76 python/tvm/relax/transform/legalize_ops/datatype.py 11 python/tvm/relax/transform/legalize_ops/distributed.py 23 python/tvm/relax/transform/legalize_ops/grad.py 171 python/tvm/relax/transform/legalize_ops/image.py 19 python/tvm/relax/transform/legalize_ops/index.py 89 python/tvm/relax/transform/legalize_ops/inspect_op.py 82 python/tvm/relax/transform/legalize_ops/linear_algebra.py 84 python/tvm/relax/transform/legalize_ops/manipulate.py 242 python/tvm/relax/transform/legalize_ops/nn.py 638 python/tvm/relax/transform/legalize_ops/qdq.py 63 python/tvm/relax/transform/legalize_ops/search.py 17 python/tvm/relax/transform/legalize_ops/statistical.py 47 python/tvm/relax/transform/legalize_ops/unary.py 42 python/tvm/relax/transform/lower_gpu_ipc_alloc_storage.py 51 python/tvm/relax/transform/optimize_layout_transform.py 36 python/tvm/relax/transform/remove_redundant_reshape.py 32 python/tvm/relax/transform/transform.py 340 python/tvm/relax/transform/tuning_api/__init__.py 3 python/tvm/relax/transform/tuning_api/_ffi_api.py 2 python/tvm/relax/transform/tuning_api/database.py 72 python/tvm/relax/transform/tuning_api/default_functions.py 155 python/tvm/relax/transform/tuning_api/primitives.py 142 python/tvm/relax/ty.py 19 python/tvm/relax/type_converter.py 78 python/tvm/relax/utils.py 206 python/tvm/relax/vm_build.py 124 python/tvm/rpc/__init__.py 4 python/tvm/rpc/_ffi_api.py 2 python/tvm/rpc/base.py 71 python/tvm/rpc/client.py 218 python/tvm/rpc/minrpc.py 26 python/tvm/rpc/proxy.py 534 python/tvm/rpc/server.py 353 python/tvm/rpc/server_ios_launcher.py 274 python/tvm/rpc/tornado_util.py 80 python/tvm/rpc/tracker.py 297 python/tvm/runtime/__init__.py 24 python/tvm/runtime/_ffi_api.py 2 python/tvm/runtime/_ffi_node_api.py 13 python/tvm/runtime/container.py 2 python/tvm/runtime/device.py 72 python/tvm/runtime/disco/__init__.py 9 python/tvm/runtime/disco/_ffi_api.py 2 python/tvm/runtime/disco/process_pool.py 105 python/tvm/runtime/disco/session.py 230 python/tvm/runtime/executable.py 48 python/tvm/runtime/module.py 279 python/tvm/runtime/ndarray.py 187 python/tvm/runtime/object.py 14 python/tvm/runtime/object_generic.py 25 python/tvm/runtime/object_path.py 79 python/tvm/runtime/packed_func.py 2 python/tvm/runtime/params.py 19 python/tvm/runtime/profiling/__init__.py 59 python/tvm/runtime/profiling/_ffi_api.py 2 python/tvm/runtime/relax_vm.py 181 python/tvm/runtime/script_printer.py 240 python/tvm/runtime/support.py 6 python/tvm/script/__init__.py 2 python/tvm/script/_ffi_api.py 2 python/tvm/script/highlight.py 176 python/tvm/script/ir_builder/__init__.py 1 python/tvm/script/ir_builder/_ffi_api.py 2 python/tvm/script/ir_builder/base.py 45 python/tvm/script/ir_builder/ir/__init__.py 13 python/tvm/script/ir_builder/ir/_ffi_api.py 2 python/tvm/script/ir_builder/ir/frame.py 5 python/tvm/script/ir_builder/ir/ir.py 38 python/tvm/script/ir_builder/relax/__init__.py 2 python/tvm/script/ir_builder/relax/_ffi_api.py 2 python/tvm/script/ir_builder/relax/distributed/__init__.py 1 python/tvm/script/ir_builder/relax/distributed/_ffi_api.py 4 python/tvm/script/ir_builder/relax/distributed/ir.py 82 python/tvm/script/ir_builder/relax/frame.py 20 python/tvm/script/ir_builder/relax/ir.py 488 python/tvm/script/ir_builder/tir/__init__.py 3 python/tvm/script/ir_builder/tir/_ffi_api.py 2 python/tvm/script/ir_builder/tir/external_kernel.py 106 python/tvm/script/ir_builder/tir/frame.py 67 python/tvm/script/ir_builder/tir/ir.py 1159 python/tvm/script/ir_builder/tir/triton.py 72 python/tvm/script/parser/__init__.py 3 python/tvm/script/parser/_core.py 4 python/tvm/script/parser/core/__init__.py 1 python/tvm/script/parser/core/diagnostics.py 138 python/tvm/script/parser/core/dispatch.py 29 python/tvm/script/parser/core/doc.py 294 python/tvm/script/parser/core/doc_core.py 800 python/tvm/script/parser/core/entry.py 72 python/tvm/script/parser/core/error.py 5 python/tvm/script/parser/core/evaluator.py 299 python/tvm/script/parser/core/parser.py 312 python/tvm/script/parser/core/utils.py 51 python/tvm/script/parser/ir/__init__.py 13 python/tvm/script/parser/ir/entry.py 17 python/tvm/script/parser/ir/parser.py 49 python/tvm/script/parser/relax/__init__.py 23 python/tvm/script/parser/relax/dist.py 73 python/tvm/script/parser/relax/entry.py 306 python/tvm/script/parser/relax/parser.py 327 python/tvm/script/parser/tir/__init__.py 11 python/tvm/script/parser/tir/entry.py 89 python/tvm/script/parser/tir/operation.py 98 python/tvm/script/parser/tir/parser.py 307 python/tvm/script/printer/__init__.py 1 python/tvm/script/printer/_ffi_api.py 2 python/tvm/script/printer/doc.py 289 python/tvm/script/printer/doc_printer.py 19 python/tvm/script/relax.py 1 python/tvm/script/tir.py 1 python/tvm/support.py 46 python/tvm/te/__init__.py 16 python/tvm/te/_ffi_api.py 2 python/tvm/te/operation.py 209 python/tvm/te/tag.py 32 python/tvm/te/tensor.py 93 python/tvm/testing/__init__.py 26 python/tvm/testing/_ffi_api.py 2 python/tvm/testing/plugin.py 207 python/tvm/testing/popen_pool.py 36 python/tvm/testing/runner.py 121 python/tvm/testing/tir.py 175 python/tvm/testing/usmp.py 12 python/tvm/testing/utils.py 949 python/tvm/tir/__init__.py 86 python/tvm/tir/_ffi_api.py 2 python/tvm/tir/analysis/__init__.py 1 python/tvm/tir/analysis/_ffi_api.py 2 python/tvm/tir/analysis/analysis.py 51 python/tvm/tir/block_dependence_info.py 25 python/tvm/tir/block_scope.py 37 python/tvm/tir/buffer.py 122 python/tvm/tir/build.py 95 python/tvm/tir/data_layout.py 38 python/tvm/tir/expr.py 473 python/tvm/tir/function.py 178 python/tvm/tir/generic.py 14 python/tvm/tir/ir_builder.py 168 python/tvm/tir/op.py 772 python/tvm/tir/pipeline.py 126 python/tvm/tir/schedule/__init__.py 7 python/tvm/tir/schedule/_ffi_api.py 2 python/tvm/tir/schedule/_type_checker.py 253 python/tvm/tir/schedule/analysis.py 37 python/tvm/tir/schedule/instruction.py 40 python/tvm/tir/schedule/schedule.py 696 python/tvm/tir/schedule/state.py 81 python/tvm/tir/schedule/trace.py 101 python/tvm/tir/schedule/transform.py 7 python/tvm/tir/stmt.py 369 python/tvm/tir/stmt_functor.py 12 python/tvm/tir/tensor_intrin/__init__.py 1 python/tvm/tir/tensor_intrin/arm_cpu.py 469 python/tvm/tir/tensor_intrin/cuda.py 1374 python/tvm/tir/tensor_intrin/dot_product_common.py 45 python/tvm/tir/tensor_intrin/hexagon.py 170 python/tvm/tir/tensor_intrin/metal.py 253 python/tvm/tir/tensor_intrin/rocm.py 371 python/tvm/tir/tensor_intrin/x86.py 74 python/tvm/tir/transform/__init__.py 2 python/tvm/tir/transform/_ffi_api.py 2 python/tvm/tir/transform/function_pass.py 48 python/tvm/tir/transform/transform.py 172 python/tvm/topi/__init__.py 23 python/tvm/topi/broadcast.py 58 python/tvm/topi/cpp/__init__.py 8 python/tvm/topi/cpp/cuda.py 2 python/tvm/topi/cpp/generic.py 2 python/tvm/topi/cpp/impl.py 2 python/tvm/topi/cpp/nn.py 2 python/tvm/topi/cpp/rocm.py 2 python/tvm/topi/cpp/utils.py 2 python/tvm/topi/cpp/vision/__init__.py 3 python/tvm/topi/cpp/vision/yolo.py 2 python/tvm/topi/cpp/x86.py 2 python/tvm/topi/einsum.py 3 python/tvm/topi/generic_op_impl.py 22 python/tvm/topi/gpu/__init__.py 2 python/tvm/topi/gpu/scan.py 366 python/tvm/topi/gpu/sort.py 590 python/tvm/topi/image/__init__.py 4 python/tvm/topi/image/dilation2d.py 87 python/tvm/topi/image/grid_sample.py 257 python/tvm/topi/image/resize.py 750 python/tvm/topi/index_put.py 55 python/tvm/topi/math.py 188 python/tvm/topi/nn/__init__.py 37 python/tvm/topi/nn/batch_matmul.py 85 python/tvm/topi/nn/batch_norm.py 61 python/tvm/topi/nn/batch_to_space_nd.py 4 python/tvm/topi/nn/bitserial_conv2d.py 160 python/tvm/topi/nn/bitserial_dense.py 43 python/tvm/topi/nn/bitserial_util.py 54 python/tvm/topi/nn/bnn.py 40 python/tvm/topi/nn/conv1d.py 27 python/tvm/topi/nn/conv1d_transpose.py 78 python/tvm/topi/nn/conv2d.py 713 python/tvm/topi/nn/conv2d_transpose.py 119 python/tvm/topi/nn/conv3d.py 61 python/tvm/topi/nn/conv3d_transpose.py 98 python/tvm/topi/nn/correlation.py 57 python/tvm/topi/nn/deformable_conv2d.py 129 python/tvm/topi/nn/dense.py 131 python/tvm/topi/nn/depth_to_space.py 39 python/tvm/topi/nn/depthwise_conv2d.py 256 python/tvm/topi/nn/dilate.py 29 python/tvm/topi/nn/elemwise.py 37 python/tvm/topi/nn/fifo_buffer.py 131 python/tvm/topi/nn/flatten.py 20 python/tvm/topi/nn/group_norm.py 3 python/tvm/topi/nn/instance_norm.py 3 python/tvm/topi/nn/layer_norm.py 3 python/tvm/topi/nn/local_response_norm.py 4 python/tvm/topi/nn/loss.py 4 python/tvm/topi/nn/lstm.py 151 python/tvm/topi/nn/mapping.py 21 python/tvm/topi/nn/pad.py 141 python/tvm/topi/nn/pixel_shuffle.py 27 python/tvm/topi/nn/pooling.py 107 python/tvm/topi/nn/qnn.py 88 python/tvm/topi/nn/rms_norm.py 3 python/tvm/topi/nn/softmax.py 86 python/tvm/topi/nn/space_to_batch_nd.py 4 python/tvm/topi/nn/space_to_depth.py 43 python/tvm/topi/nn/upsampling.py 108 python/tvm/topi/nn/utils.py 120 python/tvm/topi/nn/winograd_util.py 99 python/tvm/topi/reduction.py 40 python/tvm/topi/scan.py 102 python/tvm/topi/scatter.py 79 python/tvm/topi/scatter_elements.py 89 python/tvm/topi/searchsorted.py 53 python/tvm/topi/signal.py 116 python/tvm/topi/sort.py 81 python/tvm/topi/sparse_reshape.py 106 python/tvm/topi/tag.py 13 python/tvm/topi/tensor.py 16 python/tvm/topi/testing/__init__.py 56 python/tvm/topi/testing/adaptive_pool_python.py 84 python/tvm/topi/testing/attention_python.py 62 python/tvm/topi/testing/batch_matmul.py 21 python/tvm/topi/testing/batch_norm.py 40 python/tvm/topi/testing/batch_to_space_nd.py 39 python/tvm/topi/testing/common.py 17 python/tvm/topi/testing/conv1d_ncw_python.py 38 python/tvm/topi/testing/conv1d_transpose_ncw_python.py 37 python/tvm/topi/testing/conv2d_backcward_weight_python.py 60 python/tvm/topi/testing/conv2d_hwcn_python.py 30 python/tvm/topi/testing/conv2d_nchw_python.py 52 python/tvm/topi/testing/conv2d_nhwc_python.py 39 python/tvm/topi/testing/conv2d_transpose_python.py 75 python/tvm/topi/testing/conv3d_ncdhw_python.py 45 python/tvm/topi/testing/conv3d_ndhwc_python.py 49 python/tvm/topi/testing/conv3d_transpose_ncdhw_python.py 58 python/tvm/topi/testing/correlation_nchw_python.py 43 python/tvm/topi/testing/crop_and_resize_python.py 87 python/tvm/topi/testing/deformable_conv2d_python.py 72 python/tvm/topi/testing/dense.py 10 python/tvm/topi/testing/depth_to_space.py 15 python/tvm/topi/testing/depthwise_conv2d_python.py 62 python/tvm/topi/testing/dilate_python.py 16 python/tvm/topi/testing/gather_nd_python.py 18 python/tvm/topi/testing/gather_python.py 9 python/tvm/topi/testing/grid_sample_python.py 319 python/tvm/topi/testing/group_norm_python.py 29 python/tvm/topi/testing/instance_norm_python.py 9 python/tvm/topi/testing/l2_normalize_python.py 7 python/tvm/topi/testing/layer_norm_python.py 12 python/tvm/topi/testing/lrn_python.py 22 python/tvm/topi/testing/lstm_python.py 67 python/tvm/topi/testing/matrix_set_diag.py 33 python/tvm/topi/testing/nll_loss.py 16 python/tvm/topi/testing/one_hot.py 26 python/tvm/topi/testing/pool_grad_python.py 44 python/tvm/topi/testing/poolnd_python.py 135 python/tvm/topi/testing/reorg_python.py 25 python/tvm/topi/testing/resize_python.py 193 python/tvm/topi/testing/rms_norm_python.py 8 python/tvm/topi/testing/roi_align_python.py 132 python/tvm/topi/testing/roi_pool_python.py 39 python/tvm/topi/testing/searchsorted.py 12 python/tvm/topi/testing/sequence_mask_python.py 13 python/tvm/topi/testing/slice_axis_python.py 11 python/tvm/topi/testing/softmax_python.py 13 python/tvm/topi/testing/space_to_batch_nd.py 33 python/tvm/topi/testing/space_to_depth.py 11 python/tvm/topi/testing/strided_slice_python.py 46 python/tvm/topi/transform.py 210 python/tvm/topi/unique.py 20 python/tvm/topi/utils.py 179