![]() |
Tensor Comprehensions
|
Classes | |
| struct | SymbolTable |
| Find and categorize all variables referenced in a piece of Halide IR. More... | |
| struct | ScheduleTreeAndAccesses |
| struct | Reduction |
Typedefs | |
| typedef std::unordered_map < isl::id, Halide::Internal::Stmt, isl::IslIdIslHash > | StatementMap |
| typedef std::unordered_map < const Halide::Internal::IRNode *, isl::id > | AccessMap |
Functions | |
| SymbolTable | makeSymbolTable (const tc2halide::HalideComponents &components) |
| isl::space | makeParamSpace (isl::ctx ctx, const SymbolTable &symbolTable) |
| Make the space of all parameter values from the symbol table. More... | |
| isl::aff | makeIslAffFromInt (isl::space space, int64_t i) |
| Make a constant-valued affine function over a space. More... | |
| isl::aff | makeIslAffFromExpr (isl::space space, const Halide::Expr &e) |
| ScheduleTreeAndAccesses | makeScheduleTree (isl::space paramSpace, const Halide::Internal::Stmt &s) |
| std::vector< Reduction > | findReductions (const Halide::Internal::Stmt &s) |
| typedef std::unordered_map<const Halide::Internal::IRNode*, isl::id> tc::halide2isl::AccessMap |
| typedef std::unordered_map<isl::id, Halide::Internal::Stmt, isl::IslIdIslHash> tc::halide2isl::StatementMap |
| std::vector<Reduction> tc::halide2isl::findReductions | ( | const Halide::Internal::Stmt & | s | ) |
| isl::aff tc::halide2isl::makeIslAffFromExpr | ( | isl::space | space, |
| const Halide::Expr & | e | ||
| ) |
| isl::aff tc::halide2isl::makeIslAffFromInt | ( | isl::space | space, |
| int64_t | i | ||
| ) |
Make a constant-valued affine function over a space.
| isl::space tc::halide2isl::makeParamSpace | ( | isl::ctx | ctx, |
| const SymbolTable & | symbolTable | ||
| ) |
Make the space of all parameter values from the symbol table.
| ScheduleTreeAndAccesses tc::halide2isl::makeScheduleTree | ( | isl::space | paramSpace, |
| const Halide::Internal::Stmt & | s | ||
| ) |
Make a schedule tree from a Halide Stmt, along with auxiliary data structures describing the memory access patterns.
| SymbolTable tc::halide2isl::makeSymbolTable | ( | const tc2halide::HalideComponents & | components | ) |