Searched refs:node_by_name (Results 1 – 2 of 2) sorted by relevance
34 NodeDef* node_by_name = ctx.node_map->GetNode(node_name); in GetInputNode() local35 if (node_by_name == nullptr) { in GetInputNode()39 *node = node_by_name; in GetInputNode()
444 std::unordered_map<string, std::pair<bool, const NodeDef*>> node_by_name; in PruneGraphDefInto() local446 node_by_name[node.name()] = std::pair<bool, const NodeDef*>(false, &node); in PruneGraphDefInto()458 auto find_it = node_by_name.find(name); in PruneGraphDefInto()459 if (find_it == node_by_name.end()) { in PruneGraphDefInto()489 if (node_by_name[node.name()].first) { in PruneGraphDefInto()