Commit e2c54c4f authored by Scott Cyphers's avatar Scott Cyphers

Review comments

parent 0695fa30
...@@ -82,7 +82,7 @@ autodiff::Adjoints::Adjoints(const std::shared_ptr<Node>& y, const std::shared_p ...@@ -82,7 +82,7 @@ autodiff::Adjoints::Adjoints(const std::shared_ptr<Node>& y, const std::shared_p
// Pass 1 determines which nodes contribute to y as well as setting up a reverse // Pass 1 determines which nodes contribute to y as well as setting up a reverse
// topological sort. // topological sort.
// Number of nodes that use the a node's value // Number of nodes that use the node's value
std::unordered_map<std::shared_ptr<Node>, size_t> parent_counts; std::unordered_map<std::shared_ptr<Node>, size_t> parent_counts;
// Nodes that have been processed // Nodes that have been processed
......
...@@ -39,6 +39,7 @@ namespace ngraph ...@@ -39,6 +39,7 @@ namespace ngraph
/// view or a (possibly empty) tuple of values. /// view or a (possibly empty) tuple of values.
class Node : public std::enable_shared_from_this<Node> class Node : public std::enable_shared_from_this<Node>
{ {
// So Adjoints can call generate_adjoints
friend class autodiff::Adjoints; friend class autodiff::Adjoints;
protected: protected:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment