| 1019 | } |
| 1020 | |
| 1021 | void Edge::SetGradNode(const std::shared_ptr<GradNodeBase>& node) { |
| 1022 | VLOG(7) << "Resetting Edge(" << this << ")'s Grad Node" |
| 1023 | << " from " << (grad_node_ ? grad_node_->name() : "nullptr") << "(" |
| 1024 | << grad_node_.get() << ") to " << (node ? node->name() : "nullptr") |
| 1025 | << "(" << node.get() << ")"; |
| 1026 | grad_node_ = node; |
| 1027 | } |
| 1028 | |
| 1029 | } // namespace egr |