freetensor::StmtSetToUserGrad Struct Reference

#include <user_grad.h>

Collaboration diagram for freetensor::StmtSetToUserGrad:

Public Member Functions
	StmtSetToUserGrad (const std::unordered_set< ID > &oriStmts, const Stmt &bwdBody)

Public Attributes
std::unordered_set< ID >	oriStmts_
	Set of statements in the original program.
Stmt	bwdBody_
	Backward statement (can be a scope)

Constructor & Destructor Documentation

StmtSetToUserGrad()

freetensor::StmtSetToUserGrad::StmtSetToUserGrad ( const std::unordered_set< ID > &  oriStmts, const Stmt &  bwdBody  )

inline

Member Data Documentation

bwdBody_

Stmt freetensor::StmtSetToUserGrad::bwdBody_

Backward statement (can be a scope)

oriStmts_

std::unordered_set<ID> freetensor::StmtSetToUserGrad::oriStmts_

Set of statements in the original program.

---

The documentation for this struct was generated from the following file:

• include/autograd/user_grad.h