|
FreeTensor
|
#include <sstream>#include <analyze/all_uses.h>#include <analyze/deps.h>#include <autograd/invert_stmts.h>#include <math/parse_pb_expr.h>#include <mutator.h>#include <pass/make_nested_loops.h>#include <pass/make_reduction.h>#include <pass/replace_iter.h>
Namespaces | |
| namespace | freetensor |
Functions | |
| std::tuple< Stmt, std::unordered_map< ID, InversionInfo > > | freetensor::invertStmts (const Stmt &op, std::unordered_map< ID, std::unordered_set< ID > > *idsNeeded, std::unordered_map< StmtOrExprID, Derivative::LazyFullDerivative > *derivatives) |
| Expr cond_ |
| std::vector<IterAxis> iter_ |
| PBSet when_ |