output_intermediates.h File Reference

#include <unordered_map>
#include <unordered_set>
#include <analyze/symbol_table.h>
#include <autograd/derivative.h>
#include <mutator.h>

Include dependency graph for output_intermediates.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	freetensor::OutputIntermediates

Namespaces
namespace	freetensor

Enumerations
enum class	freetensor::OutputIntermediatesStage : int { freetensor::Forward , freetensor::Backward }

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")