Searched refs:variable_node_names (Results 1 – 1 of 1) sorted by relevance
94 std::unordered_set<string>* variable_node_names) { in GetReachableNodesAndVariables() argument113 variable_node_names->insert(node->name()); in GetReachableNodesAndVariables()192 std::unordered_set<string> variable_node_names; in FreezeGraphDef() local194 &reachable_node_names, &variable_node_names); in FreezeGraphDef()197 saved_model_bundle.session.get(), name_to_node_map, variable_node_names, in FreezeGraphDef()204 if (variable_node_names.find(node.name()) != variable_node_names.end()) { in FreezeGraphDef()208 variable_node_names.find(node.input(0)) != in FreezeGraphDef()209 variable_node_names.end()) { in FreezeGraphDef()