Searched refs:requested_grads (Results 1 – 1 of 1) sorted by relevance
538 std::unordered_map<Node*, int> requested_grads; in AddGradients() local545 ++requested_grads[nout.node()]; in AddGradients()548 for (const auto& p : requested_grads) { in AddGradients()