▼ 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 |