#include <derivative.h>

Public Member Functions
void	addPartial (const Load &x, const LazyPartialDerivative &partial)

void	setError (const std::exception_ptr &error)

bool	usingStore ()

const std::unordered_set< std::string > &	reads ()

std::vector< Stmt >	genGrads (const std::unordered_map< ID, std::string > &intermediatesMap, const std::unordered_map< StmtOrExprID, Expr > &versions, const std::unordered_map< std::string, std::string > &gradNames, const Expr &gradY)

Detailed Description

Lazy full derivative dy/dx_i for all x_i that continuously infuences y

Notes on non-differential cases:

If some x_i is missing from a LazyFullDerivative, it means x_i is not continuously infuences y, for example when x_i is an integer. So we can safely ignore all missing items.
If error_ is set, it means there is some errors or unsupported cases. It will be re-thrown if calling genGrads.

Member Function Documentation

void freetensor::Derivative::LazyFullDerivative::addPartial	(	const Load &	x,
		const LazyPartialDerivative &	partial
	)

std::vector< Stmt > freetensor::Derivative::LazyFullDerivative::genGrads	(	const std::unordered_map< ID, std::string > &	intermediatesMap,
		const std::unordered_map< StmtOrExprID, Expr > &	versions,
		const std::unordered_map< std::string, std::string > &	gradNames,
		const Expr &	gradY
	)

const std::unordered_set< std::string > & freetensor::Derivative::LazyFullDerivative::reads ( )

void freetensor::Derivative::LazyFullDerivative::setError ( const std::exception_ptr & error )

inline

bool freetensor::Derivative::LazyFullDerivative::usingStore ( )

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