freetensor::TapeStrategy Class Reference

#include <tape_strategy.h>

Public Member Functions
	TapeStrategy (const std::unordered_set< std::variant< ID, std::string, Ref< Selector > > > &tape)
	TapeStrategy (std::unordered_set< std::variant< ID, std::string, Ref< Selector > > > &&tape)
	TapeStrategy (const std::ranges::range auto &r)
	TapeStrategy (GradTapeMode mode)
TapeStrategy	alwaysTape (const std::ranges::range auto &r)
TapeStrategy	neverTape (const std::ranges::range auto &r)
std::unordered_set< ID >	getIdsToTape (const Stmt &ast) const

Constructor & Destructor Documentation

TapeStrategy() [1/4]

freetensor::TapeStrategy::TapeStrategy ( const std::unordered_set< std::variant< ID, std::string, Ref< Selector > > > &  tape )

inline

TapeStrategy() [2/4]

freetensor::TapeStrategy::TapeStrategy ( std::unordered_set< std::variant< ID, std::string, Ref< Selector > > > &&  tape )

inline

TapeStrategy() [3/4]

freetensor::TapeStrategy::TapeStrategy ( const std::ranges::range auto &  r )

inline

TapeStrategy() [4/4]

freetensor::TapeStrategy::TapeStrategy ( GradTapeMode  mode )

inline

Member Function Documentation

alwaysTape()

TapeStrategy freetensor::TapeStrategy::alwaysTape ( const std::ranges::range auto &  r )

inline

getIdsToTape()

std::unordered_set< ID > freetensor::TapeStrategy::getIdsToTape

(

const Stmt &

ast

)

const

neverTape()

TapeStrategy freetensor::TapeStrategy::neverTape ( const std::ranges::range auto &  r )

inline

---

The documentation for this class was generated from the following files:

• include/autograd/tape_strategy.h
• src/autograd/tape_strategy.cc