Searched refs:used_ops (Results 1 – 4 of 4) sorted by relevance
177 std::set<string> used_ops; // Includes both primitive ops and functions in OpsUsedByGraph() local180 const auto mark_op_as_used = [&used_ops, &functions_to_process, in OpsUsedByGraph()182 if (used_ops.insert(op).second) { in OpsUsedByGraph()204 for (const string& op_name : used_ops) { in OpsUsedByGraph()214 std::set<string> used_ops; in StrippedOpListForGraph() local215 OpsUsedByGraph(graph_def, &used_ops); in StrippedOpListForGraph()219 for (const string& op_name : used_ops) { in StrippedOpListForGraph()
277 std::set<string> used_ops; in TEST() local278 OpsUsedByGraph(graph_def, &used_ops); in TEST()279 ASSERT_EQ(std::set<string>({"B", "C"}), used_ops); in TEST()314 std::set<string> used_ops; in TEST() local315 OpsUsedByGraph(graph_def, &used_ops); in TEST()316 ASSERT_EQ(std::set<string>({"A"}), used_ops); in TEST()
155 used_ops = set() # Includes both primitive ops and functions159 if op not in used_ops and op in name_to_function:161 used_ops.add(op)170 return [op for op in used_ops if op not in name_to_function]195 used_ops = ops_used_by_graph_def(graph_def)202 for op in used_ops:207 return op_def_pb2.OpList(op=[registered_ops[op] for op in sorted(used_ops)
462 used_ops = set()465 used_ops.add(op_type)469 if op_type in used_ops: