FreeTensor
Loading...
Searching...
No Matches
Namespaces | Functions
find_tape_or_recomp_stmts.h File Reference
#include <unordered_set>
#include <autograd/derivative.h>
Include dependency graph for find_tape_or_recomp_stmts.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

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)