Searched refs:stop_backprop_nodes (Results 1 – 1 of 1) sorted by relevance
240 std::unordered_set<int> stop_backprop_nodes; in GetStopBackpropNodes() local243 stop_backprop_nodes.insert(output_node); in GetStopBackpropNodes()246 return stop_backprop_nodes; in GetStopBackpropNodes()272 std::unordered_set<int> stop_backprop_nodes = in Initialize() local305 if (stop_backprop_nodes.find(n->id()) == stop_backprop_nodes.end()) { in Initialize()