• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

Project: incubator-tvm (GitHub Link)

• incubator-tvm-master
  • .github
    • PULL_REQUEST_TEMPLATE.md
    • ISSUE_TEMPLATE.md
    • workflows
      • main.yml
    • CODEOWNERS
  • src
    • te
      • schedule
        • graph.cc
        • message_passing.cc
        • auto_inline_elem_wise.cc
        • schedule_dataflow_rewrite.cc
        • verify_compact_buffer.cc
        • schedule_postproc_rewrite_for_tensor_core.cc
        • graph.h
        • bound.cc
        • message_passing.h
        • schedule_postproc_to_primfunc.cc
        • operation_inline.cc
        • schedule_lang.cc
        • schedule_ops.cc
        • operation_inline.h
      • tensor.cc
      • autodiff
        • ad_util.h
        • ad_util.cc
        • adjoint.cc
        • jacobian.cc
      • operation
        • compute_op.cc
        • op_util.h
        • extern_op.cc
        • hybrid_op.h
        • cross_thread_reduction.cc
        • scan_op.cc
        • op_util.cc
        • tensorize.cc
        • placeholder_op.cc
        • compute_op.h
        • tensor_compute_op.cc
        • hybrid_op.cc
    • tir
      • ir
        • stmt_functor.cc
        • expr.cc
        • buffer.cc
        • function.cc
        • functor_common.h
        • transform.cc
        • stmt.cc
        • data_layout.cc
        • expr_functor.cc
      • op
        • runtime.cc
        • builtin.cc
        • op.cc
      • analysis
        • verify_gpu_code.cc
        • deep_equal.cc
        • verify_ssa.cc
        • verify_memory.cc
        • var_touch.cc
        • side_effect.cc
      • transforms
        • simplify.cc
        • storage_access.cc
        • rewrite_unsafe_select.cc
        • inject_prefetch.cc
        • lower_intrin.cc
        • ir_util.cc
        • inject_double_buffer.cc
        • vectorize_loop.cc
        • tensorcore_infer_fragment.cc
        • combine_context_call.cc
        • skip_assert.cc
        • remove_no_op.cc
        • thread_storage_sync.cc
        • unroll_loop.cc
        • split_host_device.cc
        • lower_warp_memory.cc
        • storage_flatten.cc
        • make_packed_api.cc
        • lower_custom_datatypes.cc
        • arg_binder.cc
        • storage_rewrite.cc
        • ir_util.h
        • decorate_device_scope.cc
        • remap_thread_axis.cc
        • lift_attr_scope.cc
        • storage_access.h
        • bound_checker.cc
        • inject_virtual_thread.cc
        • bf16_legalize.cc
        • lower_device_storage_access_info.cc
        • narrow_datatype.cc
        • lower_thread_allreduce.cc
        • inject_copy_intrin.cc
        • loop_partition.cc
        • arg_binder.h
        • coproc_sync.cc
        • lower_tvm_builtin.cc
    • support
      • ring_buffer.h
      • ffi_testing.cc
      • pipe.h
      • arena.h
      • base64.h
      • util.h
      • str_escape.h
      • socket.h
    • runtime
      • builtin_fp16.cc
      • crt
        • load_json.h
        • module.h
        • graph_runtime.h
        • ndarray.h
        • memory.c
        • packed_func.h
        • ndarray.c
        • crt_runtime_api.c
        • logging.h
        • graph_runtime.c
        • load_json.c
        • crt_backend_api.c
      • workspace_pool.h
      • registry.cc
      • graph
        • graph_runtime.h
        • debug
          • graph_runtime_debug.cc
        • graph_runtime.cc
      • thread_storage_scope.h
      • vm
        • memory_manager.cc
        • naive_allocator.h
        • memory_manager.h
        • profiler
          • vm.h
          • vm.cc
        • executable.cc
        • vm.cc
        • pooled_allocator.h
        • serialize_util.h
      • threading_backend.cc
      • cpu_device_api.cc
      • hexagon
        • hexagon_posix.cc
        • hexagon_module.h
        • sim
          • driver
            • pthread.h
            • sched.h
            • sim_device.cc
            • fake_pthread.cc
            • CMakeLists.txt
            • README.md
          • hexagon_device_sim.cc
          • hexagon_sim_proto.h
        • README.md
        • hexagon_device_api.cc
        • target
          • hexagon_device_target.cc
          • hexagon_stubapi.h
          • hexagon_dsprpcapi.h
          • hexagon_target_log.h
          • hexagon_stubapi.cc
          • fastrpc
            • src
              • tvm_remote_nd_imp.cc
              • tvm_hvx.cc
              • tvm_wrap_pthread.cc
              • tvm_remote_imp.cc
              • tvm_hvx.h
            • include
              • tvm_remote.idl
              • tvm_remote_nd.idl
            • CMakeLists.txt
            • README.md
          • hexagon_dsprpcapi.cc
        • hexagon_module.cc
      • module.cc
      • library_module.cc
      • meta_data.h
      • dso_library.cc
      • file_util.cc
      • object.cc
      • cuda
        • cuda_module.h
        • cuda_module.cc
        • cuda_common.h
        • cuda_device_api.cc
      • ndarray.cc
      • object_internal.h
      • metadata_module.cc
      • library_module.h
      • container.cc
      • c_runtime_api.cc
      • rocm
        • rocm_module.h
        • rocm_common.h
        • rocm_module.cc
        • rocm_device_api.cc
      • runtime_base.h
      • workspace_pool.cc
      • system_library.cc
      • file_util.h
      • vulkan
        • vulkan_module.h
        • vulkan_stream.h
        • vulkan.cc
        • vulkan_common.h
        • README.md
        • vulkan_shader.h
      • rpc
        • rpc_protocol.h
        • rpc_socket_impl.cc
        • rpc_session.h
        • rpc_event_impl.cc
        • rpc_channel.h
        • rpc_server_env.cc
        • minrpc
          • posix_popen_server.cc
          • minrpc_server.h
        • rpc_endpoint.h
        • rpc_pipe_impl.cc
        • rpc_session.cc
        • rpc_channel.cc
        • rpc_local_session.h
        • rpc_socket_impl.h
        • rpc_local_session.cc
        • rpc_endpoint.cc
        • rpc_module.cc
        • rpc_device_api.cc
      • opencl
        • aocl
          • aocl_device_api.cc
          • aocl_module.h
          • aocl_module.cc
          • aocl_common.h
        • opencl_module.h
        • opencl_device_api.cc
        • sdaccel
          • sdaccel_module.h
          • sdaccel_common.h
          • sdaccel_module.cc
          • sdaccel_device_api.cc
        • opencl_common.h
        • opencl_module.cc
      • micro
        • host_driven
          • utvm_runtime.c
          • utvm_runtime.h
          • utvm_device_dylib_redirect.c
          • utvm_runtime_enum.h
        • low_level_device.h
        • tcl_socket.h
        • openocd_low_level_device.cc
        • target_data_layout_encoder.cc
        • micro_section_allocator.h
        • micro_common.h
        • standalone
          • utvm_runtime_api.h
          • utvm_graph_runtime.cc
          • utvm_runtime_api.cc
          • utvm_graph_runtime.h
          • utvm_runtime.cc
          • README.md
          • minimal_vector.h
        • tcl_socket.cc
        • micro_session.h
        • micro_module.cc
        • micro_common.cc
        • micro_session.cc
        • target_data_layout_encoder.h
        • micro_device_api.cc
        • host_low_level_device.cc
        • device
          • riscv_spike
            • utvm_timer.c
            • utvm_init.s
          • arm
            • stm32f746xx
              • utvm_timer.c
              • utvm_init.s
          • host
            • utvm_init.c
            • utvm_timer.c
      • thread_pool.cc
      • metal
        • metal_common.h
        • metal_module.mm
        • metal_device_api.mm
        • metal_module.h
      • stackvm
        • stackvm.cc
        • stackvm_module.h
        • stackvm.h
        • stackvm_module.cc
      • contrib
        • mps
          • gemm.mm
          • mps_utils.mm
          • conv.mm
          • mps_utils.h
        • edgetpu
          • edgetpu_runtime.cc
          • edgetpu_runtime.h
        • coreml
          • coreml_runtime.mm
          • coreml_runtime.h
        • dnnl
          • dnnl.cc
          • dnnl_kernel.h
        • nnpack
          • nnpack_utils.cc
          • nnpack_utils.h
          • fully_connected.cc
          • convolution.cc
        • sort
          • sort.cc
        • tflite
          • tflite_runtime.h
          • tflite_runtime.cc
        • cublas
          • cublas_utils.h
          • cublas_utils.cc
          • cublas.cc
        • example_ext_runtime
          • example_ext_runtime.cc
        • cudnn
          • cudnn_utils.h
          • cudnn_utils.cc
          • conv_forward.cc
          • softmax.cc
        • thrust
          • thrust.cu
        • random
          • random.cc
          • mt_random_engine.cc
        • cblas
          • cblas.cc
          • gemm_common.h
        • miopen
          • miopen_utils.cc
          • conv_forward.cc
          • miopen_utils.h
        • rocblas
          • rocblas.cc
      • pack_args.h
    • ir
      • type.cc
      • env_func.cc
      • attr_functor.h
      • module.cc
      • adt.cc
      • type_functor.cc
      • expr.cc
      • tensor_type.cc
      • type_relation.cc
      • function.cc
      • transform.cc
      • error.cc
      • span.cc
      • op.cc
      • attrs.cc
    • node
      • repr_printer.cc
      • serialization.cc
      • structural_equal.cc
      • container.cc
      • structural_hash.cc
      • reflection.cc
      • attr_registry.h
    • driver
      • driver_api.cc
    • README.md
    • target
      • target_info.cc
      • source
        • literal
          • cuda_half_t.h
        • codegen_cuda.h
        • codegen_opencl.h
        • codegen_cuda.cc
        • intrin_rule_aocl.cc
        • codegen_c.h
        • codegen_source_base.h
        • intrin_rule_cuda.cc
        • codegen_c_host.cc
        • codegen_vhls.cc
        • codegen_aocl.cc
        • codegen_metal.cc
        • codegen_source_base.cc
        • intrin_rule_opencl.cc
        • intrin_rule_metal.cc
        • intrin_rule_vhls.cc
        • codegen_c_host.h
        • codegen_metal.h
        • codegen_vhls.h
        • codegen_opencl.cc
        • source_module.cc
        • codegen_c.cc
      • codegen.cc
      • target_id.cc
      • target.cc
      • datatype
        • registry.cc
        • registry.h
      • intrin_rule.h
      • spirv
        • intrin_rule_spirv.cc
        • ir_builder.h
        • ir_builder.cc
        • codegen_spirv.cc
        • codegen_spirv.h
        • build_vulkan.cc
      • intrin_rule.cc
      • opt
        • build_opencl_off.cc
        • build_metal_off.cc
        • build_rocm_off.cc
        • README
        • build_hexagon_off.cc
        • build_sdaccel_off.cc
        • build_cuda_on.cc
        • build_aocl_off.cc
        • build_cuda_off.cc
      • generic_func.cc
      • llvm
        • llvm_module.cc
        • intrin_rule_rocm.cc
        • codegen_llvm.cc
        • intrin_rule_nvptx.cc
        • codegen_amdgpu.cc
        • codegen_llvm.h
        • llvm_common.h
        • llvm_common.cc
        • codegen_x86_64.cc
        • intrin_rule_llvm.cc
        • codegen_blob.h
        • codegen_arm.cc
        • codegen_blob.cc
        • codegen_cpu.cc
        • codegen_nvptx.cc
        • codegen_cpu.h
        • intrin_rule_llvm.h
      • build_common.h
      • stackvm
        • codegen_stackvm.h
        • codegen_stackvm.cc
    • relay
      • backend
        • param_dict.h
        • vm
          • compiler.h
          • inline_primitives.cc
          • lambda_lift.cc
          • compiler.cc
          • removed_unused_funcs.cc
        • graph_runtime_codegen.cc
        • compile_engine.cc
        • interpreter.cc
        • graph_plan_memory.cc
        • utils.h
        • param_dict.cc
        • compile_engine.h
        • contrib
          • dnnl
            • codegen.cc
          • codegen_c
            • codegen.cc
            • codegen_c.h
        • build_module.cc
      • qnn
        • pass
          • legalize.cc
        • util.h
        • op
          • add.cc
          • requantize.cc
          • dense.cc
          • mul.cc
          • concatenate.cc
          • dequantize.cc
          • subtract.cc
          • quantize.cc
          • op_common.h
          • convolution.cc
        • util.cc
      • ir
        • indexed_graph.h
        • indexed_graph.cc
        • adt.cc
        • base.cc
        • expr.cc
        • function.cc
        • transform.cc
        • dataflow_matcher.cc
        • dataflow_pattern.cc
        • pattern_functor.cc
        • op_strategy.cc
        • expr_functor.cc
        • dataflow_pattern_functor.cc
      • quantize
        • annotate.cc
        • calibrate.cc
        • quantize.h
        • quantize.cc
        • realize.cc
        • partition.cc
      • op
        • debug.cc
        • type_relations.h
        • vm
          • vm.cc
        • tensor
          • transform.h
          • reduce.cc
          • unary.cc
          • binary.cc
          • transform.cc
        • nn
          • sparse.cc
          • pad.cc
          • nn.h
          • nn.cc
          • bitserial.cc
          • upsampling.cc
          • convolution.h
          • pooling.cc
          • correlation.cc
          • convolution.cc
        • type_relations.cc
        • annotation
          • annotation.cc
        • vision
          • multibox_op.cc
          • nms.cc
          • rcnn_op.cc
          • yolo.cc
        • device_copy.cc
        • dyn
          • tensor
            • transform.h
            • transform.cc
        • image
          • grid_sample.cc
          • resize.cc
          • dilation2d.cc
        • algorithm
          • argsort.cc
          • topk.cc
        • op_common.h
        • memory
          • memory.cc
      • analysis
        • mac_count.cc
        • well_formed.cc
        • type_solver.cc
        • extract_fused_functions.cc
        • feature.cc
        • call_graph.h
        • dependency_graph.h
        • annotated_region_set.h
        • annotated_region_set.cc
        • type_solver.h
        • call_graph.cc
        • dependency_graph.cc
        • match_exhaustion.cc
        • kind_check.cc
        • util.cc
      • transforms
        • partition_graph.cc
        • canonicalize_cast.cc
        • pass_util.h
        • convert_layout.cc
        • let_list.h
        • alter_op_layout.cc
        • transform_layout.h
        • combine_parallel_op_batch.h
        • combine_parallel_conv2d.cc
        • device_annotation.cc
        • type_infer.cc
        • de_duplicate.cc
        • convert_sparse_dense.cc
        • combine_parallel_op.h
        • combine_parallel_batch_matmul.cc
        • annotate_target.cc
        • simplify_fc_transpose.cc
        • legalize.cc
        • partial_eval.cc
        • simplify_inference.cc
        • fold_scale_axis.cc
        • inline.cc
        • pattern_util.h
        • expr_subst.cc
        • to_graph_normal_form.cc
        • infer_layout_util.h
        • merge_composite.cc
        • canonicalize_ops.cc
        • lazy_gradient_init.cc
        • fuse_ops.cc
        • merge_compiler_regions.cc
        • fast_math.cc
        • combine_parallel_op_batch.cc
        • to_cps.cc
        • expr_subst.h
        • gradient.cc
        • dynamic_to_static.cc
        • combine_parallel_dense.cc
        • dead_code.cc
        • forward_rewrite.cc
        • eta_expand.cc
        • combine_parallel_op.cc
        • fold_constant.cc
        • eliminate_common_subexpr.cc
        • to_a_normal_form.cc
    • printer
      • meta_data.h
      • text_printer.h
      • tir_text_printer.cc
      • doc.h
      • doc.cc
      • text_printer.cc
      • relay_text_printer.cc
    • contrib
      • hybrid
        • codegen_hybrid.h
        • codegen_hybrid.cc
      • tf_op
        • tvm_dso_op_kernels.cc
        • tvm_dso_ops.cc
    • arith
      • ir_mutator_with_analyzer.h
      • solve_linear_equation.cc
      • domain_touched.cc
      • rewrite_simplify.h
      • detect_linear_equation.cc
      • const_int_bound.cc
      • rewrite_simplify.cc
      • analyzer.cc
      • interval_set.h
      • pattern_match.h
      • int_operator.h
      • int_set.cc
      • ir_mutator_with_analyzer.cc
      • const_fold.h
      • bound_deducer.cc
      • int_constraints.cc
      • ir_visitor_with_analyzer.h
      • modular_set.cc
      • canonical_simplify.cc
    • autotvm
      • feature_visitor.h
      • feature_visitor.cc
      • touch_extractor.cc
      • touch_extractor.h
  • cmake
    • util
      • FindCUDA.cmake
      • FindROCM.cmake
      • Util.cmake
      • FindVulkan.cmake
      • FindLLVM.cmake
      • FindANTLR.cmake
    • config.cmake
    • modules
      • Vulkan.cmake
      • Metal.cmake
      • ANTLR.cmake
      • Micro.cmake
      • CUDA.cmake
      • ROCM.cmake
      • OpenMP.cmake
      • Hexagon.cmake
      • VTA.cmake
      • OpenCL.cmake
      • contrib
        • TFLite.cmake
        • HybridDump.cmake
        • Sort.cmake
        • MicroStandaloneRuntime.cmake
        • CoreML.cmake
        • Random.cmake
        • NNPack.cmake
        • DNNL.cmake
        • CODEGENC.cmake
        • TF_TVMDSOOP.cmake
        • BLAS.cmake
      • LLVM.cmake
  • CONTRIBUTORS.md
  • .asf.yaml
  • Makefile
  • include
    • tvm
      • te
        • tensor_intrin.h
        • schedule.h
        • operation.h
        • schedule_pass.h
        • tensor.h
        • autodiff.h
      • tir
        • stmt_functor.h
        • transform.h
        • data_layout.h
        • function.h
        • expr.h
        • var.h
        • analysis.h
        • builtin.h
        • op_attr_types.h
        • op.h
        • expr_functor.h
        • stmt.h
        • buffer.h
      • support
        • with.h
        • logging.h
      • runtime
        • crt
          • memory.h
        • vm.h
        • serializer.h
        • registry.h
        • c_runtime_api.h
        • module.h
        • ndarray.h
        • container.h
        • packed_func.h
        • data_type.h
        • memory.h
        • device_api.h
        • micro
          • standalone
            • utvm_runtime.h
        • threading_backend.h
        • c_backend_api.h
        • object.h
      • ir
        • type.h
        • span.h
        • adt.h
        • transform.h
        • function.h
        • tensor_type.h
        • expr.h
        • module.h
        • error.h
        • env_func.h
        • op.h
        • type_functor.h
        • type_relation.h
        • attrs.h
      • node
        • structural_equal.h
        • reflection.h
        • repr_printer.h
        • functor.h
        • serialization.h
        • attr_registry_map.h
        • container.h
        • structural_hash.h
        • node.h
      • driver
        • driver_api.h
      • target
        • codegen.h
        • generic_func.h
        • target_id.h
        • target_info.h
        • target.h
      • relay
        • type.h
        • feature.h
        • dataflow_matcher.h
        • base.h
        • adt.h
        • transform.h
        • function.h
        • dataflow_pattern_functor.h
        • expr.h
        • attrs
          • device_copy.h
          • transform.h
          • nn.h
          • debug.h
          • vision.h
          • algorithm.h
          • annotation.h
          • memory.h
          • reduce.h
          • bitserial.h
          • image.h
        • analysis.h
        • dataflow_pattern.h
        • qnn
          • transform.h
          • attrs.h
        • interpreter.h
        • op_strategy.h
        • op_attr_types.h
        • pattern_functor.h
        • op.h
        • expr_functor.h
      • arith
        • pattern.h
        • int_set.h
        • int_solver.h
        • bound.h
        • analyzer.h
  • web
    • .eslintignore
    • src
      • rpc_server.ts
      • runtime.ts
      • environment.ts
      • compact.ts
      • index.ts
      • webgpu.ts
      • memory.ts
      • types.ts
      • ctypes.ts
      • support.ts
    • jest.config.js
    • typedoc.json
    • Makefile
    • .eslintrc.json
    • emcc
      • webgpu_runtime.cc
      • tvmjs_support.cc
      • decorate_as_wasi.py
      • wasm_runtime.cc
      • preload.js
    • apps
      • browser
        • rpc_server.html
      • node
        • wasi_example.js
        • example.js
        • wasi_rpc_server.js
    • rollup.config.js
    • README.md
    • package.json
    • tests
      • node
        • test_ndarray.js
        • test_packed_func.js
        • test_module_load.js
      • python
        • webgpu_rpc_test.py
        • prepare_test_libs.py
        • websock_rpc_test.py
    • tsconfig.json
    • .gitignore
  • licenses
    • LICENSE.blockingconcurrentqueue.txt
    • LICENSE.concurrentqueue.txt
    • LICENSE.builtin_fp16.txt
    • LICENSE.bfloat16.txt
    • LICENSE.picojson.txt
    • LICENSE.cma.txt
    • LICENSE.dlpack.txt
    • LICENSE.rang.txt
  • jvm
    • pom.xml
    • README.md
    • native
      • src
        • main
          • native
            • org_apache_tvm_native_c_api.cc
            • jni_helper_func.h
      • linux-x86_64
        • pom.xml
      • osx-x86_64
        • pom.xml
      • pom.xml
    • assembly
      • linux-x86_64
        • src
          • main
            • assembly
              • assembly.xml
        • pom.xml
      • osx-x86_64
        • src
          • main
            • assembly
              • assembly.xml
        • pom.xml
      • pom.xml
    • core
      • src
        • main
          • java
            • org
              • apache
                • tvm
                  • TVMValueHandle.java
                  • TVMValueString.java
                  • TVMValueBytes.java
                  • LibInfo.java
                  • NDArrayBase.java
                  • APIInternal.java
                  • Module.java
                  • TVMValueDouble.java
                  • Base.java
                  • NativeLibraryLoader.java
                  • Function.java
                  • TVMContext.java
                  • TVMValue.java
                  • ArgTypeCode.java
                  • NDArray.java
                  • API.java
                  • rpc
                    • Server.java
                    • Client.java
                    • SocketChannel.java
                    • RPCSession.java
                    • TVMRemoteContext.java
                    • RPCWatchdog.java
                    • ConnectProxyServerProcessor.java
                    • ConnectTrackerServerProcessor.java
                    • Utils.java
                    • NativeServerLoop.java
                    • RPC.java
                    • StandaloneServerProcessor.java
                    • ServerProcessor.java
                  • TVMValueLong.java
                  • TVMType.java
                  • contrib
                    • GraphRuntime.java
                    • GraphModule.java
                  • TVMValueNull.java
        • test
          • java
            • org
              • apache
                • tvm
                  • ModuleTest.java
                  • TestUtils.java
                  • FunctionTest.java
                  • rpc
                    • RPCTest.java
                  • NDArrayTest.java
                  • contrib
                    • GraphRuntimeTest.java
          • scripts
            • test_rpc_proxy_server.py
            • test_add_gpu.py
            • test_graph_runtime.py
            • test_add_cpu.py
      • pom.xml
    • conf
      • log4j.properties
      • google_checks.xml
  • 3rdparty
    • dlpack
    • picojson
      • picojson.h
      • README.md
    • compiler-rt
      • builtin_fp16.h
    • bfloat16
      • bfloat16.cc
    • rang
    • cma
      • settings.mk
      • cma_api_impl.h
      • cma.h
    • vta-hw
    • dmlc-core
  • .gitmodules
  • version.py
  • LICENSE
  • .clang-format
  • vta
    • runtime
      • device_api.cc
      • runtime.h
      • runtime.cc
    • tutorials
      • matrix_multiply.py
      • vta_get_started.py
      • README.txt
      • frontend
        • deploy_detection.py
        • deploy_classification.py
        • README.txt
      • autotvm
        • tune_relay_vta.py
        • README.txt
      • optimize
        • convolution_opt.py
        • matrix_multiply_opt.py
        • README.txt
    • python
      • vta
        • rpc_client.py
        • libinfo.py
        • transform.py
        • top
          • util.py
          • vta_group_conv2d.py
          • graphpack.py
          • bitpack.py
          • vta_dense.py
          • op.py
          • vta_conv2d.py
          • __init__.py
          • vta_conv2d_transpose.py
        • bitstream.py
        • __init__.py
        • program_bitstream.py
        • intrin.py
        • build_module.py
        • environment.py
        • testing
          • util.py
          • simulator.py
          • __init__.py
        • exec
          • __init__.py
          • rpc_server.py
    • README.md
    • tests
      • python
        • integration
          • test_benchmark_topi_group_conv2d.py
          • test_benchmark_topi_conv2d_transpose.py
          • test_benchmark_topi_dense.py
          • test_benchmark_gemm.py
          • test_benchmark_topi_conv2d.py
        • pynq
          • test_program_rpc.py
        • de10nano
          • test_program_rpc.py
        • unittest
          • test_environment.py
          • test_vta_insn.py
    • scripts
      • tune_resnet.py
      • tune_conv2d_transpose.py
      • tune_group_conv2d.py
      • tune_dense.py
      • tune_conv2d.py
  • topi
    • src
      • schedule.cc
      • nn.cc
      • elemwise.cc
      • reduction.cc
      • transform.cc
      • broadcast.cc
      • vision.cc
    • include
      • topi
        • transform.h
        • tags.h
        • nn.h
        • cuda
          • normalization.h
          • dense.h
          • pooling.h
          • injective.h
          • reduction.h
          • softmax.h
        • util.h
        • nn
          • dense.h
          • local_response_norm.h
          • bnn.h
          • pooling.h
          • mapping.h
          • flatten.h
          • batch_matmul.h
          • softmax.h
          • bias_add.h
          • dilate.h
        • elemwise.h
        • rocm
          • normalization.h
          • dense.h
          • pooling.h
          • injective.h
          • reduction.h
          • softmax.h
        • x86
          • bnn.h
          • injective.h
          • default.h
        • generic
          • extern.h
          • injective.h
          • default.h
        • detail
          • fuse.h
          • tensor_utils.h
          • constant_utils.h
          • extern.h
          • pad_utils.h
          • ravel_unravel.h
          • broadcast.h
          • array_utils.h
        • vision
          • reorg.h
        • reduction.h
        • contrib
          • cublas.h
          • rocblas.h
        • broadcast.h
    • recipe
      • gemm
        • gemm_int8.py
        • cuda_gemm_square.py
        • android_gemm_square.py
      • reduce
        • test_reduce_map.py
      • broadcast
        • test_broadcast_map.py
      • rnn
        • lstm.py
        • matexp.py
      • conv
        • depthwise_conv2d_test.py
        • test_conv_int8_arm.py
        • test_conv_int8_intel.py
        • test_conv2d_hwcn_map.py
    • python
      • topi
        • cpp
          • nn.py
          • util.py
          • cuda.py
          • impl.py
          • generic.py
          • __init__.py
          • vision
            • yolo.py
            • __init__.py
          • rocm.py
          • x86.py
        • bifrost
          • depthwise_conv2d.py
          • __init__.py
          • dense.py
          • transforms.py
          • conv2d.py
          • gemm.py
        • broadcast.py
        • sparse
          • csrmv.py
          • csrmm.py
          • __init__.py
          • dense.py
        • util.py
        • generic_op_impl.py
        • transform.py
        • cuda
          • conv2d_transpose_nchw.py
          • injective.py
          • conv2d_nhwc.py
          • nn.py
          • conv3d_winograd.py
          • dense_tensorcore.py
          • conv3d_ndhwc_tensorcore.py
          • depthwise_conv2d.py
          • conv3d_direct.py
          • conv2d_hwcn.py
          • rcnn
            • proposal.py
            • __init__.py
          • batch_matmul.py
          • conv2d_direct.py
          • correlation.py
          • conv3d_alter_op.py
          • softmax.py
          • deformable_conv2d.py
          • vision.py
          • conv1d.py
          • conv3d_transpose_ncdhw.py
          • conv3d.py
          • group_conv2d_nchw.py
          • conv2d_winograd.py
          • conv2d_int8.py
          • conv2d_nhwc_winograd.py
          • pooling.py
          • conv2d_alter_op.py
          • __init__.py
          • tensor_intrin.py
          • conv1d_transpose_ncw.py
          • sort.py
          • conv2d_nhwc_tensorcore.py
          • dense.py
          • nms.py
          • sparse.py
          • conv2d.py
          • ssd
            • multibox.py
            • __init__.py
          • reduction.py
        • argwhere.py
        • nn
          • util.py
          • bitserial_util.py
          • depthwise_conv2d.py
          • flatten.py
          • fifo_buffer.py
          • batch_matmul.py
          • mapping.py
          • bnn.py
          • correlation.py
          • space_to_depth.py
          • softmax.py
          • conv3d_transpose.py
          • local_response_norm.py
          • deformable_conv2d.py
          • conv1d.py
          • pad.py
          • conv3d.py
          • elemwise.py
          • conv1d_transpose.py
          • dilate.py
          • pooling.py
          • winograd_util.py
          • __init__.py
          • dense.py
          • bitserial_conv2d.py
          • bitserial_dense.py
          • sparse.py
          • depth_to_space.py
          • conv2d.py
          • conv2d_transpose.py
          • upsampling.py
        • intel_graphics
          • depthwise_conv2d.py
          • conv2d_alter_op.py
          • __init__.py
          • conv2d.py
        • rocm
          • nn.py
          • __init__.py
          • dense.py
          • conv2d.py
        • x86
          • roi_align.py
          • injective.py
          • nn.py
          • util.py
          • depthwise_conv2d.py
          • batch_matmul.py
          • binarize_pack.py
          • conv3d_transpose.py
          • binary_dense.py
          • conv1d.py
          • conv3d.py
          • conv2d_avx_1x1.py
          • conv2d_int8.py
          • pooling.py
          • conv2d_alter_op.py
          • __init__.py
          • tensor_intrin.py
          • conv2d_avx_common.py
          • dense.py
          • bitserial_conv2d.py
          • bitserial_dense.py
          • sparse.py
          • conv2d.py
          • reduction.py
          • conv2d_transpose.py
        • generic
          • injective.py
          • nn.py
          • image.py
          • search.py
          • default.py
          • vision.py
          • extern.py
          • __init__.py
          • sort.py
          • conv2d.py
        • scatter.py
        • mali
          • depthwise_conv2d.py
          • __init__.py
          • dense.py
          • conv2d.py
        • tag.py
        • __init__.py
        • sort.py
        • hls
          • injective.py
          • nn.py
          • __init__.py
        • vision
          • rcnn
            • roi_align.py
            • roi_pool.py
            • proposal.py
            • __init__.py
          • __init__.py
          • nms.py
          • reorg.py
          • ssd
            • multibox.py
            • __init__.py
        • image
          • grid_sample.py
          • resize.py
          • __init__.py
          • dilation2d.py
        • tensor.py
        • math.py
        • reduction.py
        • testing
          • conv2d_nhwc_python.py
          • roi_pool_python.py
          • bilinear_resize_python.py
          • conv2d_nchw_python.py
          • softmax_python.py
          • pool1d_python.py
          • gather_nd_python.py
          • slice_axis_python.py
          • roi_align_python.py
          • batch_matmul.py
          • deformable_conv2d_nchw_python.py
          • space_to_depth.py
          • conv2d_hwcn_python.py
          • dilate_python.py
          • conv2d_transpose_python.py
          • gather_python.py
          • conv3d_ndhwc_python.py
          • crop_and_resize_python.py
          • depthwise_conv2d_python.py
          • conv3d_transpose_ncdhw_python.py
          • sequence_mask_python.py
          • one_hot.py
          • strided_slice_python.py
          • conv1d_ncw_python.py
          • upsampling_python.py
          • pool_grad_python.py
          • conv1d_transpose_ncw_python.py
          • common.py
          • __init__.py
          • grid_sample_python.py
          • adaptive_pool_python.py
          • l2_normalize_python.py
          • reorg_python.py
          • correlation_nchw_python.py
          • conv3d_ncdhw_python.py
          • depth_to_space.py
          • trilinear_resize3d_python.py
          • pool3d_python.py
          • lrn_python.py
        • arm_cpu
          • injective.py
          • conv2d_spatial_pack.py
          • depthwise_conv2d.py
          • cortex_m7
            • conv2d
              • direct.py
              • direct_simd.py
              • __init__.py
            • __init__.py
            • micro_kernel
              • __init__.py
              • gemm.py
          • conv2d_gemm.py
          • conv2d_int8.py
          • conv2d_alter_op.py
          • __init__.py
          • tensor_intrin.py
          • bitserial_conv2d.py
          • bitserial_dense.py
          • conv2d.py
          • conv2d_transpose.py
      • setup.py
    • README.md
    • tests
      • python
        • test_topi_bitserial_conv2d.py
        • test_topi_conv2d_winograd.py
        • test_topi_bnn.py
        • test_topi_conv2d_nhwc_winograd.py
        • test_topi_sparse.py
        • test_topi_group_conv2d.py
        • test_topi_relu.py
        • test_topi_tensor.py
        • test_topi_conv2d_NCHWc.py
        • test_topi_transform.py
        • test_topi_softmax.py
        • test_topi_dilate.py
        • test_topi_conv3d_ndhwc_tensorcore.py
        • test_topi_group_conv2d_NCHWc_int8.py
        • test_topi_conv2d_nhwc_tensorcore.py
        • test_topi_conv3d_ndhwc.py
        • test_topi_batch_matmul.py
        • test_topi_reduce.py
        • test_topi_depthwise_conv2d_back_input.py
        • test_topi_conv3d_ncdhw.py
        • test_topi_image.py
        • test_topi_deformable_conv2d.py
        • test_topi_conv3d_winograd.py
        • test_topi_conv2d_int8.py
        • test_topi_dense.py
        • test_topi_depthwise_conv2d.py
        • test_topi_depthwise_conv2d_back_weight.py
        • test_topi_sort.py
        • test_topi_broadcast.py
        • test_topi_lrn.py
        • test_fifo_buffer.py
        • test_topi_vision.py
        • test_topi_conv2d_nchw.py
        • test_topi_pooling.py
        • test_topi_correlation.py
        • test_topi_bitserial_conv2d_rasp.py
        • test_topi_depth_to_space.py
        • test_topi_conv3d_transpose_ncdhw.py
        • common.py
        • test_topi_conv2d_nhwc.py
        • test_topi_bitserial_dense.py
        • test_topi_conv2d_hwcn.py
        • test_topi_clip.py
        • test_topi_conv1d.py
        • test_topi_util.py
        • test_topi_basic.py
        • test_topi_reorg.py
        • test_topi_upsampling.py
        • test_topi_conv1d_transpose_ncw.py
        • test_topi_conv2d_nhwc_pack_int8.py
        • test_topi_math.py
        • test_topi_dense_tensorcore.py
        • test_topi_conv2d_transpose_nchw.py
        • test_topi_space_to_depth.py
        • test_topi_matmul.py
  • apps
    • vta_rpc
      • start_rpc_server.sh
      • start_rpc_server_to_tracker.sh
    • android_deploy
      • gradle.properties
      • dev_tools
        • sign_apk.sh
        • gen_keystore.sh
      • build.gradle
      • README.md
      • app
        • src
          • main
            • res
              • xml
                • provider_paths.xml
              • layout
                • content_main.xml
                • activity_main.xml
              • values
                • styles.xml
                • colors.xml
                • strings.xml
            • AndroidManifest.xml
            • jni
              • make
                • config.mk
              • Application.mk
              • Android.mk
              • tvm_runtime.h
              • build.sh
            • java
              • org
                • apache
                  • tvm
                    • android
                      • demo
                        • MainActivity.java
        • download-models.gradle
        • build.gradle
        • .gitignore
      • settings.gradle
      • .gitignore
    • sgx
      • src
        • main.rs
        • build_model.py
      • Cargo.toml
      • Cargo.lock
      • .rustfmt.toml
      • read_results.py
      • README.md
      • .cargo
        • config
      • build.rs
    • tf_tvmdsoop
      • prepare_and_test_tfop_module.sh
      • CMakeLists.txt
      • tests
        • test_tfop_module.py
    • lldb
      • dot.lldbinit.in
      • tvm.py
    • extension
      • src
        • tvm_ext.cc
      • Makefile
      • python
        • tvm_ext
          • __init__.py
      • README.md
      • tests
        • test_ext.py
      • .gitignore
    • android_rpc
      • dev_tools
        • sign_apk.sh
        • gen_keystore.sh
      • build.gradle
      • README.md
      • tests
        • android_rpc_test.py
      • app
        • src
          • main
            • res
              • layout
                • activity_rpc.xml
                • content_main.xml
                • activity_main.xml
                • content_rpc.xml
              • mipmap-hdpi
              • values
                • styles.xml
                • colors.xml
                • strings.xml
              • mipmap-mdpi
            • AndroidManifest.xml
            • jni
              • make
                • config.mk
              • Application.mk
              • Android.mk
              • tvm_runtime.h
              • build.sh
            • java
              • org
                • apache
                  • tvm
                    • tvmrpc
                      • MainActivity.java
                      • RPCProcessor.java
                      • RPCActivity.java
        • build.gradle
        • .gitignore
      • settings.gradle
      • .gitignore
    • cpp_rpc
      • main.cc
      • rpc_server.h
      • Makefile
      • rpc_tracker_client.h
      • win32_process.cc
      • rpc_env.h
      • win32_process.h
      • rpc_env.cc
      • rpc_server.cc
      • CMakeLists.txt
      • README.md
    • android_camera
      • gradle.properties
      • dev_tools
        • sign_apk.sh
        • gen_keystore.sh
      • models
        • prepare_model.py
      • build.gradle
      • README.md
      • app
        • src
          • main
            • res
              • xml
                • provider_paths.xml
              • layout
                • fragment_camera2_basic.xml
                • activity_main.xml
                • listview_row.xml
              • values
                • styles.xml
                • colors.xml
                • strings.xml
              • drawable
                • item_selector.xml
            • rs
              • yuv420888.rs
            • AndroidManifest.xml
            • jni
              • make
                • config.mk
              • Application.mk
              • Android.mk
              • tvm_runtime.h
              • build.sh
            • java
              • org
                • apache
                  • tvm
                    • android
                      • androidcamerademo
                        • MainActivity.java
                        • Camera2BasicFragment.java
        • build.gradle
      • settings.gradle
      • .gitignore
    • dso_plugin_module
      • Makefile
      • plugin_module.cc
      • README.md
      • test_plugin_module.py
      • .gitignore
    • README.md
    • bundle_deploy
      • Makefile
      • runtime.c
      • demo.cc
      • bundle.c
      • build_model.py
      • test_static.c
      • bundle.cc
      • bundle.h
      • bundle_static.c
      • README.md
      • runtime.cc
      • test.cc
      • demo_static.c
    • ios_rpc
      • tvmrpc.xcodeproj
        • project.pbxproj
        • project.xcworkspace
          • contents.xcworkspacedata
      • tvmrpcLauncher
        • Info.plist
        • tvmrpcLauncher.mm
      • README.md
      • tests
        • ios_rpc_mobilenet.py
        • ios_rpc_test.py
      • .gitignore
      • tvmrpc
        • Base.lproj
          • Main.storyboard
          • LaunchScreen.storyboard
        • ViewController.h
        • Info.plist
        • TVMRuntime.h
        • Assets.xcassets
          • AppIcon.appiconset
            • Contents.json
        • ViewController.mm
        • AppDelegate.m
        • main.m
        • AppDelegate.h
        • TVMRuntime.mm
      • init_proj.py
    • howto_deploy
      • python_deploy.py
      • Makefile
      • prepare_test_libs.py
      • cpp_deploy.cc
      • tvm_runtime_pack.cc
      • run_example.sh
      • README.md
    • benchmark
      • util.py
      • mobile_gpu_imagenet_bench.py
      • README.md
      • gpu_imagenet_bench.py
      • arm_cpu_imagenet_bench.py
    • rocm_rpc
      • Makefile
      • rocm_runtime_pack.cc
      • start_rpc_server.sh
      • README.md
  • docker
    • with_the_same_user
    • Dockerfile.ci_wasm
    • Dockerfile.demo_gpu
    • Dockerfile.ci_gpu
    • Dockerfile.conda_cuda100
    • Dockerfile.ci_lint
    • Dockerfile.demo_android
    • Dockerfile.ci_i386
    • bash.sh
    • Dockerfile.conda_cpu
    • install
      • ubuntu_install_java.sh
      • ubuntu1804_install_python.sh
      • ubuntu_install_llvm.sh
      • ubuntu_install_dnnl.sh
      • ubuntu_install_core.sh
      • ubuntu1804_install_clang_format.sh
      • ubuntu_install_rocm.sh
      • ubuntu_install_nodejs.sh
      • ubuntu_install_tflite.sh
      • ubuntu_install_tensorflow.sh
      • ubuntu_install_vulkan.sh
      • ubuntu_install_emscripten.sh
      • ubuntu_install_gradle.sh
      • ubuntu1804_install_llvm.sh
      • ubuntu_install_antlr.sh
      • ubuntu_install_mxnet.sh
      • ubuntu_install_sphinx.sh
      • ubuntu_install_coreml.sh
      • ubuntu_install_onnx.sh
      • ubuntu_install_python_package.sh
      • ubuntu_install_rust.sh
      • ubuntu_install_dgl.sh
      • ubuntu_install_nnpack.sh
      • ubuntu_install_golang.sh
      • ubuntu_install_caffe2.sh
      • install_tvm_cpu.sh
      • ubuntu_install_opencl.sh
      • install_tvm_gpu.sh
      • ubuntu_install_redis.sh
      • ubuntu_install_chisel.sh
      • ubuntu_install_darknet.sh
      • ubuntu_install_gluoncv.sh
      • ubuntu_install_rat.sh
      • ubuntu_install_androidsdk.sh
      • ubuntu_install_python.sh
    • Dockerfile.demo_cpu
    • Dockerfile.conda_cuda90
    • README.md
    • Dockerfile.ci_jekyll
    • Dockerfile.demo_opencl
    • Dockerfile.ci_cpu
    • build.sh
  • NEWS.md
  • tutorials
    • relay_quick_start.py
    • language
      • tensorize.py
      • schedule_primitives.py
      • intrin_math.py
      • tuple_inputs.py
      • README.txt
      • scan.py
      • tedd.py
      • extern_op.py
      • reduction.py
    • cross_compilation_and_rpc.py
    • README.txt
    • topi
      • README.txt
      • intro_topi.py
    • dev
      • low_level_custom_pass.py
      • relay_pass_infra.py
      • README.txt
    • frontend
      • from_tflite.py
      • from_onnx.py
      • deploy_quantized.py
      • from_coreml.py
      • deploy_model_on_rasp.py
      • from_caffe2.py
      • deploy_model_on_android.py
      • deploy_sparse.py
      • README.txt
      • from_tensorflow.py
      • from_darknet.py
      • using_external_lib.py
      • deploy_prequantized.py
      • build_gcn.py
      • from_mxnet.py
      • deploy_prequantized_tflite.py
      • from_keras.py
      • deploy_ssd_gluoncv.py
      • from_pytorch.py
    • micro
      • README.txt
      • micro_tflite.py
    • autotvm
      • tune_simple_template.py
      • tune_relay_x86.py
      • README.txt
      • tune_relay_cuda.py
      • tune_relay_mobile_gpu.py
      • tune_conv2d_cuda.py
      • tune_relay_arm.py
    • optimize
      • opt_gemm.py
      • README.txt
      • opt_conv_cuda.py
      • opt_conv_tensorcore.py
      • opt_matmul_auto_tensorcore.py
    • tensor_expr_get_started.py
  • python
    • tvm
      • testing.py
      • te
        • operation.py
        • autodiff.py
        • hybrid
          • module.py
          • util.py
          • preprocessor.py
          • __init__.py
          • parser.py
          • calls.py
          • runtime.py
        • tag.py
        • __init__.py
        • tensor_intrin.py
        • schedule.py
        • _ffi_api.py
        • tensor.py
      • _ffi
        • _cython
          • ndarray.pxi
          • base.pxi
          • packed_func.pxi
          • core.pyx
          • object.pxi
        • libinfo.py
        • _pyversion.py
        • _cy3
          • __init__.py
        • registry.py
        • runtime_ctypes.py
        • __init__.py
        • _ctypes
          • ndarray.py
          • object.py
          • __init__.py
          • packed_func.py
          • types.py
        • _cy2
          • __init__.py
        • base.py
      • tir
        • stmt_functor.py
        • stmt.py
        • op.py
        • generic.py
        • buffer.py
        • function.py
        • analysis
          • analysis.py
          • __init__.py
          • _ffi_api.py
        • ir_builder.py
        • __init__.py
        • data_layout.py
        • expr.py
        • _ffi_api.py
        • transform
          • function_pass.py
          • transform.py
          • __init__.py
          • _ffi_api.py
      • runtime
        • vm.py
        • module.py
        • container.py
        • profiler_vm.py
        • ndarray.py
        • _ffi_node_api.py
        • object_generic.py
        • object.py
        • __init__.py
        • packed_func.py
        • _ffi_api.py
      • ir
        • _ffi_transform_api.py
        • module.py
        • type.py
        • container.py
        • tensor_type.py
        • attrs.py
        • transform.py
        • op.py
        • json_compact.py
        • function.py
        • __init__.py
        • adt.py
        • expr.py
        • type_relation.py
        • _ffi_api.py
        • base.py
      • generic.py
      • driver
        • __init__.py
        • build_module.py
      • __init__.py
      • rpc
        • client.py
        • server.py
        • minrpc.py
        • tornado_util.py
        • __init__.py
        • tracker.py
        • proxy.py
        • _ffi_api.py
        • base.py
      • micro
        • __init__.py
        • base.py
        • device
          • arm
            • stm32f746xx.py
            • __init__.py
          • riscv_spike.py
          • host.py
          • __init__.py
          • base.py
      • error.py
      • target
        • datatype.py
        • target.py
        • codegen.py
        • __init__.py
        • generic_func.py
        • _ffi_api.py
        • arm_isa.py
        • intrin.py
      • relay
        • expr_functor.py
        • ty.py
        • loops.py
        • param_dict.py
        • type_functor.py
        • _parser.py
        • backend
          • vm.py
          • _vm.py
          • graph_runtime_codegen.py
          • interpreter.py
          • __init__.py
          • _backend.py
          • compile_engine.py
        • std
          • gradient.rly
          • prelude.rly
          • core.rly
        • qnn
          • transform.py
          • op
            • legalizations.py
            • op.py
            • _make.py
            • __init__.py
            • layout_conversions.py
            • qnn.py
          • __init__.py
        • scope_builder.py
        • grammar
          • Relay.g4
          • __init__.py
          • py3
            • RelayLexer.py
            • .gitattributes
            • RelayParser.py
            • __init__.py
            • RelayVisitor.py
          • .gitignore
        • quantize
          • quantize.py
          • kl_divergence.py
          • _annotate.py
          • _calibrate.py
          • __init__.py
          • _quantize.py
          • _partition.py
        • op
          • vm
            • vm.py
            • __init__.py
            • _ffi_api.py
          • _algorithm.py
          • transform.py
          • algorithm.py
          • _reduce.py
          • _tensor_grad.py
          • nn
            • nn.py
            • util.py
            • _nn.py
            • _make.py
            • __init__.py
          • op.py
          • _make.py
          • __init__.py
          • annotation
            • annotation.py
            • _make.py
            • __init__.py
          • vision
            • _vision.py
            • multibox.py
            • yolo.py
            • _make.py
            • __init__.py
            • rcnn.py
            • _rcnn.py
            • _yolo.py
            • nms.py
          • reduce.py
          • _tensor.py
          • dyn
            • _make.py
            • __init__.py
            • _transform.py
          • image
            • _image.py
            • image.py
            • _make.py
            • __init__.py
          • contrib
            • coreml.py
            • dnnl.py
            • register.py
            • __init__.py
          • tensor.py
          • _transform.py
          • op_attrs.py
          • strategy
            • hls.py
            • cuda.py
            • bifrost.py
            • arm_cpu.py
            • intel_graphics.py
            • generic.py
            • mali.py
            • __init__.py
            • rocm.py
            • x86.py
          • memory
            • memory.py
            • _make.py
            • __init__.py
        • data_dep_optimization
          • bsr_dense.py
          • simplify_fc_transpose.py
          • __init__.py
          • utils.py
        • function.py
        • _make.py
        • analysis
          • sparse_dense.py
          • analysis.py
          • annotated_regions.py
          • __init__.py
          • call_graph.py
          • _ffi_api.py
          • feature.py
        • __init__.py
        • adt.py
        • dataflow_pattern
          • __init__.py
          • _ffi.py
        • parser.py
        • frontend
          • keras.py
          • qnn_torch.py
          • mxnet.py
          • caffe2.py
          • pytorch.py
          • coreml.py
          • darknet.py
          • tflite.py
          • mxnet_qnn_op_utils.py
          • tflite_flexbuffer.py
          • onnx.py
          • common.py
          • __init__.py
          • tensorflow_parser.py
          • nnvm_common.py
          • tensorflow.py
        • expr.py
        • prelude.py
        • debug.py
        • _ffi_api.py
        • _build_module.py
        • transform
          • transform.py
          • __init__.py
          • memory_plan.py
          • _ffi_api.py
          • memory_alloc.py
        • base.py
        • build_module.py
        • testing
          • init.py
          • vgg.py
          • yolo_detection.py
          • inception_v3.py
          • config.py
          • dcgan.py
          • mobilenet.py
          • squeezenet.py
          • resnet_3d.py
          • temp_op_attr.py
          • darknet.py
          • resnet.py
          • lstm.py
          • layers.py
          • densenet.py
          • py_converter.py
          • __init__.py
          • dqn.py
          • tf.py
          • mlp.py
          • nat.py
      • contrib
        • rocblas.py
        • tflite_runtime.py
        • coreml_runtime.py
        • util.py
        • mxnet.py
        • cudnn.py
        • dlpack.py
        • download.py
        • random.py
        • emcc.py
        • clang.py
        • miopen.py
        • graph_runtime.py
        • cblas.py
        • peak.py
        • tf_op
          • module.py
          • __init__.py
        • pickle_memoize.py
        • mps.py
        • binutil.py
        • cublas.py
        • xcode.py
        • __init__.py
        • tedd.py
        • cc.py
        • nvcc.py
        • rocm.py
        • target
          • coreml.py
          • __init__.py
        • rpc.py
        • sparse.py
        • tar.py
        • sdaccel.py
        • spirv.py
        • cublaslt.py
        • nnpack.py
        • debugger
          • debug_runtime.py
          • debug_result.py
          • __init__.py
        • ndk.py
      • arith
        • bound.py
        • int_solver.py
        • pattern.py
        • __init__.py
        • _ffi_api.py
        • int_set.py
        • analyzer.py
      • autotvm
        • env.py
        • measure
          • measure_methods.py
          • measure.py
          • executor.py
          • local_executor.py
          • __init__.py
        • graph_tuner
          • dynamic_programming_tuner.py
          • utils
            • traverse_graph.py
            • __init__.py
            • utils.py
          • base_graph_tuner.py
          • pbqp_tuner.py
          • __init__.py
          • dynamic_programming_stage.py
          • _base.py
        • util.py
        • tuner
          • ga_tuner.py
          • metric.py
          • xgboost_tuner.py
          • tuner.py
          • index_based_tuner.py
          • callback.py
          • model_based_tuner.py
          • __init__.py
          • xgboost_cost_model.py
          • sa_model_optimizer.py
        • task
          • relay_integration.py
          • task.py
          • code_hash.py
          • space.py
          • topi_integration.py
          • __init__.py
          • dispatcher.py
        • tophub.py
        • database.py
        • __init__.py
        • record.py
        • feature.py
      • exec
        • rpc_proxy.py
        • query_rpc_tracker.py
        • rpc_tracker.py
        • measure_peak.py
        • autotvm_log_editor.py
        • __init__.py
        • rpc_server.py
    • setup.py
    • .gitignore
  • DISCLAIMER
  • nnvm
    • src
      • c_api
        • c_api_error.cc
        • c_api_common.h
        • c_api_graph.cc
        • c_api_symbolic.cc
      • pass
        • order_mutation.cc
        • graph_algorithm.h
        • print_graph_ir.cc
        • plan_memory.cc
        • gradient.cc
        • infer_shape_type.cc
        • correct_layout.cc
        • saveload_json.cc
        • place_device.cc
      • README.md
      • core
        • graph.cc
        • node.cc
        • pass.cc
        • symbolic.cc
        • op.cc
    • Makefile
    • include
      • nnvm
        • graph_attr_types.h
        • base.h
        • c_api.h
        • tuple.h
        • symbolic.h
        • pass_functions.h
        • graph.h
        • op_attr_types.h
        • node.h
        • pass.h
        • op.h
        • layout.h
    • make
      • config.mk
    • amalgamation
      • Makefile
      • README
      • amalgamation.py
      • generate.py
      • .gitignore
    • README.md
    • tests
      • cpp
        • unittest.mk
        • op_test.cc
        • .gitignore
        • tuple_test.cc
  • golang
    • src
      • value_test.go
      • util.go
      • value.go
      • function.go
      • gotvm.cc
      • error_test.go
      • tvm_runtime_pack.cc
      • gotvm.h
      • bytearray_test.go
      • function_test.go
      • ndarray.go
      • module.go
      • module_test.go
      • type.go
      • array_test.go
      • error.go
      • gotvm_test.go
      • gotvm.go
      • bytearray.go
      • context.go
    • sample
      • pack_func_convert.go
      • pack_func_register.go
      • Makefile
      • pack_func_closure_return.go
      • deploy.py
      • complex.go
      • gen_mobilenet_lib.py
      • pack_func_closure_arg.go
      • pack_func_handle_arg.go
      • simple.go
    • Makefile
    • README.md
  • CMakeLists.txt
  • README.md
  • tests
    • cpp
      • build_module_test.cc
      • target_test.cc
      • crt_memory_test.cc
      • expr_test.cc
      • arith_simplify_test.cc
      • tir_analysis_side_effect.cc
      • tensor_test.cc
      • topi_ewise_test.cc
      • relay_transform_sequential_test.cc
      • pattern_match_test.cc
      • object_protocol_test.cc
      • relay_build_module_test.cc
      • packed_func_test.cc
      • container_test.cc
      • ir_functor_test.cc
      • utvm_runtime_standalone_test.cc
      • attrs_test.cc
      • threading_backend_test.cc
      • .gitignore
      • relay_pass_type_infer_test.cc
    • python
      • nightly
        • quantization
          • test_quantization_accuracy.py
      • integration
        • test_scan.py
        • test_winograd_nnpack.py
        • test_tuning.py
        • test_reduce.py
        • test_ewise_fpga.py
        • test_gemm.py
        • test_ewise.py
        • test_dot.py
      • frontend
        • coreml
          • model_zoo
            • __init__.py
          • test_forward.py
        • onnx
          • test_forward.py
        • mxnet
          • test_graph.py
          • test_qnn_ops_utils.py
          • model_zoo
            • vgg.py
            • inception_v3.py
            • dcgan.py
            • squeezenet.py
            • resnet.py
            • __init__.py
            • dqn.py
            • mlp.py
          • test_forward.py
        • tflite
          • test_forward.py
        • caffe2
          • test_graph.py
          • model_zoo
            • squeezenet.py
            • __init__.py
          • test_forward.py
        • pytorch
          • qnn_test.py
          • lstm_test.py
          • test_forward.py
        • test_common.py
        • keras
          • test_forward.py
        • darknet
          • test_forward.py
        • tensorflow
          • test_no_op.py
          • test_debugging.py
          • test_control_flow.py
          • test_bn_dynamic.py
          • test_forward.py
      • relay
        • test_pass_partition_graph.py
        • test_op_grad_level4.py
        • test_ir_structural_equal_hash.py
        • test_any.py
        • test_op_level2.py
        • test_type_functor.py
        • test_pass_remove_unused_functions.py
        • test_op_qnn_concatenate.py
        • test_pass_fast_math.py
        • test_call_graph.py
        • test_op_grad_level1.py
        • test_pass_lambda_lift.py
        • test_memory_passes.py
        • test_pass_auto_quantize.py
        • test_pass_mac_count.py
        • test_adt.py
        • test_op_level3.py
        • test_op_qnn_mul.py
        • test_cpp_build_module.py
        • test_pass_eta_expand.py
        • test_param_dict.py
        • test_pass_dynamic_to_static.py
        • test_pass_eliminate_common_subexpr.py
        • test_pass_partial_eval.py
        • test_op_qnn_requantize.py
        • test_dataflow_pattern.py
        • test_external_runtime.py
        • test_pass_combine_parallel_conv2d.py
        • test_pass_qnn_legalize.py
        • test_expr_functor.py
        • test_typecall.py
        • test_op_qnn_dense.py
        • test_op_qnn_dequantize.py
        • test_pass_manager.py
        • test_op_level5.py
        • test_backend_compile_engine.py
        • test_vm_serialization.py
        • test_op_fast_math.py
        • test_pass_annotation.py
        • test_change_batch.py
        • test_pass_lazy_gradient_init.py
        • test_pass_to_graph_normal_form.py
        • test_op_grad_level2.py
        • test_pass_combine_parallel_dense.py
        • test_ir_well_formed.py
        • test_op_level6.py
        • test_pass_check_kind.py
        • test_pass_simplify_inference.py
        • test_type_solver.py
        • test_pass_fold_scale_axis.py
        • test_pass_vars.py
        • benchmarking
          • benchmark_vm.py
        • test_ir_nodes.py
        • test_ir_parser.py
        • test_op_qnn_quantize.py
        • test_ir_text_printer.py
        • test_op_qnn_add.py
        • test_analysis_feature.py
        • test_pass_merge_compiler_regions.py
        • test_op_level4.py
        • test_pass_unmatched_cases.py
        • test_error_reporting.py
        • test_simplify_fc_transpose.py
        • test_pass_legalize.py
        • test_pass_inline.py
        • test_autotvm_task_extraction.py
        • test_ir_bind.py
        • test_pass_combine_parallel_batch_matmul.py
        • test_cmp_op.py
        • test_op_level10.py
        • test_pass_annotate_target.py
        • test_backend_interpreter.py
        • test_pass_to_a_normal_form.py
        • test_json_compact.py
        • test_pass_to_cps.py
        • test_backend_graph_runtime.py
        • test_op_level1.py
        • test_op_qnn_subtract.py
        • test_pass_convert_op_layout.py
        • test_op_grad_level3.py
        • test_pass_merge_composite.py
        • test_debug.py
        • test_sparse_dense_convert.py
        • test_pass_gradient.py
        • test_py_converter.py
        • test_pass_dead_code_elimination.py
        • test_analysis_extract_fused_functions.py
        • test_annotated_regions.py
        • test_pass_canonicalize_cast.py
        • test_op_qnn_conv2d.py
        • dyn
          • test_dynamic_op_level3.py
        • test_ir_module.py
        • test_pass_fuse_ops.py
        • test_vm.py
        • test_pass_fold_constant.py
        • test_ir_op.py
        • test_pass_alter_op_layout.py
        • test_external_codegen.py
        • test_type_infer.py
        • test_op_grad_level10.py
      • contrib
        • test_dlpack.py
        • test_random.py
        • test_gemm_acc16.py
        • test_sparse.py
        • test_coreml_runtime.py
        • test_gemm_acc32_vnni.py
        • test_cudnn.py
        • test_binutil.py
        • test_mps.py
        • test_rpc_proxy.py
        • test_util.py
        • test_miopen.py
        • test_mxnet_bridge.py
        • test_edgetpu_runtime.py
        • test_cublas.py
        • test_nnpack.py
        • test_rpc_tracker.py
        • test_tedd.py
        • test_rocblas.py
        • test_coreml_codegen.py
        • test_tflite_runtime.py
        • test_sort.py
        • test_cblas.py
      • unittest
        • test_tir_transform_bf16_legalize.py
        • test_target_codegen_bool.py
        • test_tir_transform_loop_partition.py
        • test_arith_canonical_simplify.py
        • test_target_codegen_device.py
        • test_arith_modular_set.py
        • test_autotvm_flop_calculator.py
        • test_target_codegen_rocm.py
        • test_tir_transform_rewrite_unsafe_select.py
        • test_tir_structural_equal_hash.py
        • test_tir_transform_remove_no_op.py
        • test_runtime_container.py
        • test_runtime_graph.py
        • test_target_codegen_extern.py
        • test_tir_analysis_verify_memory.py
        • test_autotvm_database.py
        • test_te_autodiff.py
        • test_tir_transform_lower_intrin.py
        • test_ir_attrs.py
        • test_te_build_lower.py
        • test_target_codegen_vm_basic.py
        • test_te_tensor.py
        • test_te_tag.py
        • test_tir_intrin.py
        • test_runtime_micro.py
        • test_autotvm_dispatch_context.py
        • test_target_codegen_c_host.py
        • test_autotvm_measure.py
        • test_format_si_prefix.py
        • test_autotvm_index_tuner.py
        • test_target_codegen_x86.py
        • test_runtime_vm_profiler.py
        • test_runtime_ndarray.py
        • test_target_codegen_arm.py
        • test_runtime_extension.py
        • test_tir_data_layout.py
        • test_tir_nodes.py
        • test_tir_buffer.py
        • test_runtime_error.py
        • test_target_codegen_opencl.py
        • test_tir_transform_simplify.py
        • test_tir_transform_vectorize.py
        • test_tir_analysis_expr_deep_equal.py
        • test_tir_ops.py
        • test_runtime_measure.py
        • test_arith_solve_linear_system.py
        • test_target_target.py
        • test_tir_stmt_functor_ir_transform.py
        • test_target_codegen_cuda.py
        • test_tir_transform_make_packed_api.py
        • test_tir_transform_combine_context_call.py
        • test_te_schedule_graph.py
        • test_tir_transform_storage_rewrite.py
        • test_tir_analysis_verify_gpu_code.py
        • test_target_custom_datatypes.py
        • test_runtime_packed_func.py
        • test_arith_domain_touched.py
        • test_runtime_module_load.py
        • test_autotvm_record.py
        • test_node_reflection.py
        • test_tir_transform_instrument_bound_checkers.py
        • test_te_schedule_bound_inference.py
        • test_te_schedule_postproc_rewrite_for_tensor_core.py
        • test_ir_type.py
        • test_autotvm_graph_tuner_utils.py
        • test_tir_transform_decorate_device_scope.py
        • test_tir_transform_inject_double_buffer.py
        • test_te_schedule_tensorize.py
        • test_te_group.py
        • test_arith_detect_linear_equation.py
        • test_testing.py
        • test_te_hybrid_script.py
        • test_tir_ir_builder.py
        • test_tir_transform_inject_copy_intrin.py
        • test_ir_container.py
        • test_runtime_rpc.py
        • test_te_tensor_overload.py
        • test_autotvm_common.py
        • test_tir_transform_storage_flatten.py
        • test_tir_transform_prim_func_pass.py
        • test_target_codegen_static_init.py
        • test_target_codegen_cross_llvm.py
        • test_runtime_heterogeneous.py
        • test_te_schedule_lstm.py
        • test_runtime_graph_debug.py
        • test_te_schedule_ops.py
        • test_runtime_module_export.py
        • test_arith_deduce_bound.py
        • test_te_verify_compute.py
        • test_autotvm_executor.py
        • test_tir_constructor.py
        • test_tir_transform_inject_virtual_thread.py
        • test_tir_analysis_usedef.py
        • test_arith_detect_clip_bound.py
        • test_tir_transform_lower_warp_memory.py
        • test_te_schedule_bound_inference_tiling.py
        • test_te_schedule_tensor_core.py
        • test_arith_intset.py
        • test_arith_rewrite_simplify.py
        • test_autotvm_xgboost_model.py
        • test_tir_transform_unroll_loop.py
        • test_target_codegen_vulkan.py
        • test_tir_transform_coproc_sync.py
        • test_autotvm_feature.py
        • test_target_codegen_blob.py
        • test_tir_transform_narrow_datatype.py
        • test_tir_analysis_verify_ssa.py
        • test_autotvm_graph_tuner_core.py
        • test_tir_transform_thread_sync.py
        • test_tir_transform_lift_attr_scope.py
        • test_autotvm_space.py
        • test_arith_const_int_bound.py
        • test_target_codegen_llvm.py
        • test_te_schedule.py
    • azure-pipelines
      • main.yml
    • lint
      • add_asf_header.py
      • check_file_type.py
      • pylintrc
      • git-clang-format.sh
      • rat-excludes
    • scripts
      • task_cpp_unittest.sh
      • task_config_build_cpu.sh
      • task_lint.sh
      • task_golang.sh
      • setup-pytest-env.sh
      • task_config_build_gpu_vulkan.sh
      • task_sphinx_precheck.sh
      • task_config_build_wasm.sh
      • task_rust.sh
      • task_python_vta_fsim.sh
      • task_python_vta_tsim.sh
      • task_java_unittest.sh
      • task_python_nightly.sh
      • task_config_build_i386.sh
      • task_python_topi.sh
      • task_web_wasm.sh
      • task_build.sh
      • task_python_integration.sh
      • task_config_build_gpu.sh
      • task_python_frontend.sh
      • task_clean.sh
      • task_python_frontend_cpu.sh
      • task_python_unittest.sh
      • task_python_docs.sh
    • micro
      • test_runtime_micro_on_arm.py
  • conda
    • conda_build_config.yaml
    • cross-linux.cmake
    • build_cpu.sh
    • tvm
      • meta.yaml
      • build.sh
    • build_cuda.sh
    • tvm-libs
      • meta.yaml
      • build.sh
    • render_cuda.py
    • Dockerfile.template
  • rust
    • tvm-graph-rt
      • src
        • module
          • mod.rs
          • syslib.rs
          • dso.rs
        • array.rs
        • graph.rs
        • allocator.rs
        • errors.rs
        • threading.rs
        • workspace.rs
        • lib.rs
      • Cargo.toml
      • tests
        • test_wasm32
          • src
            • main.rs
            • build_test_lib.py
          • Cargo.toml
          • .cargo
            • config
          • build.rs
        • build_model.py
        • test_graph_serde.rs
        • test_tvm_dso
          • src
            • main.rs
            • build_test_lib.py
          • Cargo.toml
          • build.rs
        • test_tvm_basic
          • src
            • main.rs
            • build_test_lib.py
          • Cargo.toml
          • build.rs
        • test_nn
          • src
            • main.rs
            • build_test_graph.py
          • Cargo.toml
          • build.rs
        • .gitignore
    • Cargo.toml
    • tvm-rt
      • src
        • string.rs
        • object
          • object_ptr.rs
          • mod.rs
        • module.rs
        • ndarray.rs
        • to_boxed_fn.rs
        • array.rs
        • to_function.rs
        • context.rs
        • value.rs
        • errors.rs
        • function.rs
        • lib.rs
      • Cargo.toml
      • README.md
      • .gitignore
    • tvm
      • src
        • runtime
          • mod.rs
        • ir
          • mod.rs
          • relay
            • mod.rs
        • transform.rs
        • lib.rs
      • Cargo.toml
      • .travis.yml
      • README.md
      • .gitignore
    • common
      • src
        • packed_func.rs
        • array.rs
        • value.rs
        • errors.rs
        • lib.rs
      • Cargo.toml
      • build.rs
    • tvm-sys
      • src
        • packed_func.rs
        • datatype.rs
        • array.rs
        • context.rs
        • value.rs
        • errors.rs
        • byte_array.rs
        • lib.rs
      • Cargo.toml
      • build.rs
    • runtime
      • src
        • module
          • mod.rs
          • syslib.rs
          • dso.rs
        • array.rs
        • graph.rs
        • allocator.rs
        • errors.rs
        • threading.rs
        • workspace.rs
        • lib.rs
      • Cargo.toml
      • .travis.yml
      • tests
        • test_wasm32
          • src
            • main.rs
            • build_test_lib.py
          • Cargo.toml
          • .cargo
            • config
          • build.rs
        • build_model.py
        • test_graph_serde.rs
        • test_tvm_dso
          • src
            • main.rs
            • build_test_lib.py
          • Cargo.toml
          • build.rs
        • test_tvm_basic
          • src
            • main.rs
            • build_test_lib.py
          • Cargo.toml
          • build.rs
        • test_nn
          • src
            • main.rs
            • build_test_graph.py
          • Cargo.toml
          • build.rs
        • .gitignore
    • .rustfmt.toml
    • frontend
      • src
        • module.rs
        • ndarray.rs
        • context.rs
        • value.rs
        • errors.rs
        • function.rs
        • lib.rs
      • Cargo.toml
      • examples
        • resnet
          • src
            • main.rs
            • build_resnet.py
          • Cargo.toml
          • README.md
          • build.rs
      • .travis.yml
      • README.md
      • tests
        • callback
          • src
            • bin
              • int.rs
              • string.rs
              • error.rs
              • float.rs
              • array.rs
          • Cargo.toml
        • basics
          • src
            • tvm_add.py
            • main.rs
          • Cargo.toml
          • .gitignore
          • build.rs
      • .gitignore
    • .gitignore
    • tvm-macros
      • src
        • object.rs
        • util.rs
        • import_module.rs
        • external.rs
        • lib.rs
      • Cargo.toml
    • macros
      • src
        • lib.rs
      • Cargo.toml
  • KEYS
  • NOTICE
  • .gitignore
  • Jenkinsfile
  • docs
    • Makefile
    • contribute
      • git_howto.rst
      • code_guide.rst
      • document.rst
      • committer_guide.rst
      • release_process.rst
      • code_review.rst
      • community.rst
      • pull_request.rst
      • error_handling.rst
      • index.rst
    • faq.rst
    • Doxyfile
    • api
      • links.rst
      • python
        • rpc.rst
        • graph_runtime.rst
        • runtime.rst
        • te.rst
        • micro.rst
        • driver.rst
        • target.rst
        • ndarray.rst
        • contrib.rst
        • ir.rst
        • vta
          • index.rst
        • tir.rst
        • index.rst
        • error.rst
        • autotvm.rst
        • relay
          • nn.rst
          • vision.rst
          • frontend.rst
          • transform.rst
          • backend.rst
          • image.rst
          • dataflow_pattern.rst
          • testing.rst
          • index.rst
          • analysis.rst
        • topi.rst
    • deploy
      • cpp_deploy.rst
      • hls.rst
      • integrate.rst
      • android.rst
      • index.rst
    • README.txt
    • vta
      • install.rst
      • dev
        • config.rst
        • hardware.rst
        • index.rst
      • index.rst
      • .gitignore
    • dev
      • debugger.rst
      • convert_layout.rst
      • runtime.rst
      • relay_op_strategy.rst
      • hybrid_script.rst
      • relay_intro.rst
      • relay_bring_your_own_codegen.rst
      • security.rst
      • relay_add_pass.rst
      • virtual_machine.rst
      • introduction_to_module_serialization.rst
      • benchmark.rst
      • relay_add_op.rst
      • relay_pass_infra.rst
      • inferbound.rst
      • codebase_walkthrough.rst
      • index.rst
    • install
      • from_source.rst
      • docker.rst
      • nnpack.rst
      • index.rst
    • frontend
      • tensorflow.rst
    • index.rst
    • conf.py
    • langref
      • relay_type.rst
      • hybrid_script.rst
      • relay_adt.rst
      • relay_expr.rst
      • index.rst
      • relay_op.rst
      • relay_pattern.rst
    • genindex.rst
    • .gitignore
    • _static
      • img
        • README
      • css
        • tvm_theme.css

Open Deep Learning Compiler Stack

Documentation | Contributors | Community | Release Notes

Apache TVM (incubating) is a compiler stack for deep learning systems. It is designed to close the gap between the productivity-focused deep learning frameworks, and the performance- and efficiency-focused hardware backends. TVM works with deep learning frameworks to provide end to end compilation to different backends.

License

© Contributors Licensed under an Apache-2.0 license.

Contribute to TVM

TVM adopts apache committer model, we aim to create an open source project that is maintained and owned by the community. Checkout the Contributor Guide

Acknowledgement

We learned a lot from the following projects when building TVM.

• Halide: Part of TVM's TIR and arithmetic simplification module originates from Halide. We also learned and adapted some part of lowering pipeline from Halide.
• Loopy: use of integer set analysis and its loop transformation primitives.
• Theano: the design inspiration of symbolic scan operator for recurrence.