Contents

struct

SushiAI::Autograd::Provenance

Records, for every node differentiate() appended, the forward node it serves.

Declared in
include/SushiAI/autograd/provenance.hpp

See also

include/SushiAI/eager/README.md

Public attributes

std::size_t forward_node_count = 0

Node count of the graph before differentiation; the first backward id.

std::vector< Graph::NodeId > origin

Origins in append order: origin[i] belongs to node forward_node_count + i.

Public member functions

Graph::NodeId origin_of(Graph::NodeId backward_node) const

Returns the forward node that backward_node was emitted for.

Parameters

backward_node

A node id at or above forward_node_count.