|
tvm
|
Directories | |
| directory | schedule |
Files | |
| file | analysis.h [code] |
| Analysis utilities and passes for TIR. | |
| file | block_dependence_info.h [code] |
| Define BlockDependenceInfoNode that uses the BlockScope and StmtSRef objects to store the block level dependences. | |
| file | block_scope.h [code] |
| Definition of two pillar data structure for TensorIR scheduling: StmtSRef, BlockScope. | |
| file | buffer.h [code] |
| Symbolic n-dimensional array, to represent a memory buffer. | |
| file | builtin.h [code] |
| TIR builtin intrinsics. | |
| file | data_layout.h [code] |
| Layout expression to describe the data organization of a tensor. And BijectiveLayout to mapping two data layouts between each other. | |
| file | data_type_rewriter.h [code] |
| Rewrite the data type of expressions. | |
| file | expr.h [code] |
| TIR expressions. | |
| file | expr_functor.h [code] |
| Functors for tir expressions. | |
| file | function.h [code] |
| TIR Function. | |
| file | index_map.h [code] |
| Defines a remapping of buffer indices. | |
| file | op.h [code] |
| Common operators defined for Expr. | |
| file | op_attr_types.h [code] |
| Attribute types in the Op registry for TIR ops. | |
| file | stmt.h [code] |
| TIR statements. | |
| file | stmt_functor.h [code] |
| Functors for tir stmts utility functions to call common functors. | |
| file | transform.h [code] |
| TIR specific transformation passes. | |
| file | utils.h [code] |
| file | var.h [code] |
| Variables in the TIR. | |