Searched refs:variable_node_names (Results 1 – 1 of 1) sorted by relevance
90 std::unordered_set<string>* variable_node_names) { in GetReachableNodesAndVariables() argument109 variable_node_names->insert(node->name()); in GetReachableNodesAndVariables()188 std::unordered_set<string> variable_node_names; in FreezeGraphDef() local190 &reachable_node_names, &variable_node_names); in FreezeGraphDef()193 saved_model_bundle.session.get(), name_to_node_map, variable_node_names, in FreezeGraphDef()200 if (variable_node_names.find(node.name()) != variable_node_names.end()) { in FreezeGraphDef()204 variable_node_names.find(node.input(0)) != in FreezeGraphDef()205 variable_node_names.end()) { in FreezeGraphDef()