Searched refs:used_ops (Results 1 – 3 of 3) sorted by relevance
214 std::set<string> used_ops; // Includes both primitive ops and functions in OpsUsedByGraph() local217 const auto mark_op_as_used = [&used_ops, &functions_to_process, in OpsUsedByGraph()219 if (used_ops.insert(op).second) { in OpsUsedByGraph()241 for (const string& op_name : used_ops) { in OpsUsedByGraph()251 std::set<string> used_ops; in StrippedOpListForGraph() local252 OpsUsedByGraph(graph_def, &used_ops); in StrippedOpListForGraph()256 for (const string& op_name : used_ops) { in StrippedOpListForGraph()
322 std::set<string> used_ops; in TEST() local323 OpsUsedByGraph(graph_def, &used_ops); in TEST()324 ASSERT_EQ(std::set<string>({"B", "C"}), used_ops); in TEST()359 std::set<string> used_ops; in TEST() local360 OpsUsedByGraph(graph_def, &used_ops); in TEST()361 ASSERT_EQ(std::set<string>({"A"}), used_ops); in TEST()
156 used_ops = set() # Includes both primitive ops and functions160 if op not in used_ops and op in name_to_function:162 used_ops.add(op)176 return [op for op in used_ops if op not in name_to_function]196 used_ops = ops_used_by_graph_def(graph_def)198 for op in sorted(used_ops):