output_intermediates.cc File Reference

#include <analyze/symbol_table.h>
#include <autograd/analyze_version.h>
#include <autograd/output_intermediates.h>

Include dependency graph for output_intermediates.cc:

Namespaces
namespace	freetensor

Functions
std::tuple< Stmt, std::unordered_map< ID, std::string >, std::unordered_map< StmtOrExprID, Expr >, std::unordered_map< ID, Expr >, std::unordered_set< ID >, std::unordered_map< std::string, std::pair< std::string, Expr > > >	freetensor::outputIntermediates (const Stmt &op, const std::unordered_map< ID, std::unordered_set< ID > > &needVersions, const std::unordered_map< StmtOrExprID, Derivative::LazyFullDerivative > &derivatives, OutputIntermediatesStage stage=OutputIntermediatesStage::Forward, const std::string &varSuffix=".tape")
Stmt	freetensor::outputAllIntermedaites (const Stmt &op, const std::unordered_set< ID > &intermediates, OutputIntermediatesStage stage=OutputIntermediatesStage::Forward, const std::string &varSuffix=".tape")