#include <analyze/all_defs.h>#include <analyze/deps.h>#include <analyze/symbol_table.h>#include <autograd/find_tape_or_recomp_stmts.h>
Include dependency graph for find_tape_or_recomp_stmts.cc:
Namespaces | |
| namespace | freetensor |
Functions | |
| std::pair< std::unordered_map< ID, std::unordered_set< ID > >, std::unordered_map< ID, std::unordered_set< ID > > > | freetensor::findTapeOrRecompStmts (const Stmt &op, const std::unordered_set< ID > &defsToTape, const std::unordered_set< ID > defsNeedGrad, std::unordered_map< StmtOrExprID, Derivative::LazyFullDerivative > &derivatives) |
Variable Documentation
◆ ids_
| std::unordered_set<ID> ids_ |
