| ▼ tvm |
|
| ▼ arith |
|
| analyzer.h |
Algebra expression simplifications |
| bound.h |
Bound deducers |
| int_set.h |
Integer set |
| int_solver.h |
Integer constraints data structures and solvers |
| pattern.h |
Expression pattern detectors |
| util.h |
Utils for arithmetic analysis |
| ▼ driver |
|
| driver_api.h |
Compiler driver APIs to drive the compilation |
| ▼ ir |
|
| adt.h |
Algebraic data type definitions |
| attrs.h |
Helpers for attribute objects |
| env_func.h |
Serializable global function used in IR |
| error.h |
Utilities for error tracking and reporting |
| expr.h |
Base expr nodes in TVM |
| function.h |
Function nodes |
| module.h |
IRModule that holds the functions and type definitions |
| op.h |
Primitive operators(builtin intrinsics) and registry for them |
| span.h |
Span information for debugging purposes |
| tensor_type.h |
Polymorphic tensor types |
| transform.h |
|
| type.h |
IR/AST nodes for the unified type system in TVM |
| type_functor.h |
A way to defined arbitrary function signature with dispatch on types |
| type_relation.h |
Type relation and function for type inference(checking) |
| ▼ node |
|
| attr_registry_map.h |
Attribute map used in registry |
| container.h |
Array/Map container in the DSL graph |
| functor.h |
Defines the Functor data structures |
| node.h |
Definitions and helper macros for IR/AST nodes |
| reflection.h |
Reflection and serialization of compiler IR/AST nodes |
| repr_printer.h |
Printer class to print repr string of each AST/IR nodes |
| serialization.h |
|
| structural_equal.h |
Structural equality comparison |
| structural_hash.h |
|
| ▼ relay |
|
| ▼ attrs |
|
| algorithm.h |
|
| annotation.h |
Attribute for annotation operators |
| bitserial.h |
Auxiliary attributes for bitserial operators |
| debug.h |
Auxiliary attributes for debug operators |
| device_copy.h |
Attribute for the device copy operator |
| image.h |
Auxiliary attributes for image operators |
| memory.h |
Attributes for memory operators |
| nn.h |
Auxiliary attributes for nn operators |
| reduce.h |
Auxiliary attributes for reduce operators |
| transform.h |
Transform operators |
| vision.h |
Auxiliary attributes for vision operators |
| ▼ qnn |
|
| attrs.h |
Auxiliary attributes for qnn operators |
| transform.h |
|
| adt.h |
Algebraic data types for Relay |
| analysis.h |
The set of Relay analysis passes written in C++ |
| base.h |
Base classes for the Relay IR |
| dataflow_matcher.h |
A pattern matcher for matching dataflow properties |
| dataflow_pattern.h |
A pattern language for matching dataflow properties |
| dataflow_pattern_functor.h |
A set of passes for operating on pattern graphs |
| expr.h |
Relay expression language |
| expr_functor.h |
A more powerful visitor which enables defining arbitrary function signatures with type based dispatch on first argument |
| feature.h |
Detect features used in Expr/Module |
| function.h |
Relay Function |
| interpreter.h |
An interpreter for Relay |
| op.h |
Primitive operators(builtin intrinsics) |
| op_attr_types.h |
The Expr and related elements in DataFlow construction |
| op_strategy.h |
The Relay operator Strategy and related data structure |
| pattern_functor.h |
A more powerful visitor on ADT patterns that enables defining arbitrary function signatures with type-based dispatch on first argument |
| transform.h |
Relay specific transformation passes |
| type.h |
Relay typed AST nodes |
| ▼ runtime |
|
| ▼ crt |
|
| memory.h |
The virtual memory manager for micro-controllers |
| ▼ micro |
|
| ▼ standalone |
|
| utvm_runtime.h |
|
| c_backend_api.h |
TVM runtime backend API |
| c_runtime_api.h |
|
| container.h |
Common POD(plain old data) container types |
| data_type.h |
|
| device_api.h |
Abstract device memory management API |
| memory.h |
Runtime memory management |
| module.h |
Runtime container of the functions generated by TVM, This is used to support dynamically link, load and save functions from different convention under unified API |
| ndarray.h |
A device-independent managed NDArray abstraction |
| object.h |
A managed object in the TVM runtime |
| packed_func.h |
Type-erased function used across TVM API |
| registry.h |
This file defines the TVM global function registry |
| serializer.h |
Serializer extension to support TVM data types Include this file to enable serialization of DLDataType, DLContext |
| threading_backend.h |
Utilities for manipulating thread pool threads |
| vm.h |
A virtual machine for executing Relay programs |
| ▼ support |
|
| with.h |
RAII wrapper function to enter and exit a context object similar to python’s with syntax |
| ▼ target |
|
| codegen.h |
Translates IRModule to runtime::Module |
| generic_func.h |
Generic function that can be specialzied on a per target basis |
| target.h |
Compilation target object |
| target_info.h |
Various information about target |
| ▼ te |
|
| autodiff.h |
Automatic differentiation of tensor expressions |
| operation.h |
Operation node can generate one or multiple Tensors |
| schedule.h |
Define a schedule |
| schedule_pass.h |
Collection of Schedule pass functions |
| tensor.h |
Dataflow tensor object |
| tensor_intrin.h |
Tensor intrinsic operations |
| ▼ tir |
|
| analysis.h |
Analysis utilitie and passes for TIR |
| buffer.h |
Symbolic n-dimensional array, to represent a memory buffer |
| data_layout.h |
Layout expression to describe the data organization of a tensor. And BijectiveLayout to mapping two data layouts between each other |
| expr.h |
TIR expressions |
| expr_functor.h |
Functors for tir expressions |
| function.h |
TIR Function |
| op.h |
Common operators defined for Expr |
| stmt.h |
TIR statements |
| stmt_functor.h |
Functors for tir stmts utility functions to call common functors |
| transform.h |
TIR specific transformation passes |
| var.h |
Variables in the TIR |
| ▼ topi |
|
| ▼ include |
|
| ▼ topi |
|
| ▼ contrib |
|
| cublas.h |
External function interface to cuBLAS libraries |
| rocblas.h |
|
| ▼ cuda |
|
| dense.h |
CUDA schedule for dense operation |
| injective.h |
CUDA schedule for injective operations |
| normalization.h |
CUDA schedule for LRN and l2 normalization operations |
| pooling.h |
CUDA schedule for pooling operations |
| reduction.h |
CUDA schedule for reduction operations |
| softmax.h |
|
| ▼ detail |
|
| array_utils.h |
Utility functions for handling arrays |
| broadcast.h |
Detail broadcast |
| constant_utils.h |
Utility functions for handling constants in TVM expressions |
| extern.h |
Helpers for using external functions |
| fuse.h |
Fuse operation |
| pad_utils.h |
Padding helpers |
| ravel_unravel.h |
Index ravel and unraval operations |
| tensor_utils.h |
Utility functions for handling tensor |
| ▼ generic |
|
| default.h |
Generic default schedule |
| extern.h |
Schedule for extern followed by injective ops |
| injective.h |
Generic schedule for injective operations |
| ▼ nn |
|
| batch_matmul.h |
Batch matmul op constructions |
| bias_add.h |
Bias_add op constructions |
| bnn.h |
Binary op constructions |
| dense.h |
Dense op constructions |
| dilate.h |
Dilate op constructions |
| flatten.h |
Softmax op constructions |
| local_response_norm.h |
Local response normalization op constructions |
| mapping.h |
Mapping op constructions |
| pooling.h |
Pooling op constructions |
| softmax.h |
Softmax op constructions |
| ▼ rocm |
|
| dense.h |
Rocm schedule for dense operation |
| injective.h |
Rocm schedule for injective operations |
| normalization.h |
Rocm schedule for LRN and l2 normalization operations |
| pooling.h |
Rocm schedule for pooling operations |
| reduction.h |
Rocm schedule for reduction operations |
| softmax.h |
|
| ▼ vision |
|
| reorg.h |
Reorg op constructions |
| ▼ x86 |
|
| bnn.h |
X86 schedule for binary operations |
| default.h |
Default x86 schedule |
| injective.h |
X86 schedule for injective ops |
| broadcast.h |
Broadcast op constructions |
| elemwise.h |
Elementwise op constructions |
| nn.h |
NN op constructions |
| reduction.h |
Reduction op constructors |
| tags.h |
External function interface to rocBLAS libraries |
| transform.h |
Transform op constructors |
| util.h |
Topi utility function |