Home
last modified time | relevance | path

Searched refs:source_ops (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dmerge_control_flow.cc80 llvm::SmallSetVector<Operation*, 4> source_ops; in SafeToMerge() local
81 source_ops.insert(source); in SafeToMerge()
83 source_ops.insert(&op); in SafeToMerge()
86 source_ops.insert(&op); in SafeToMerge()
94 if (!source_ops.contains(user)) dependencies.push_back(user); in SafeToMerge()
98 if (!source_ops.contains(successor)) dependencies.push_back(successor); in SafeToMerge()
103 if (!source_ops.contains(successor)) dependencies.push_back(successor); in SafeToMerge()
116 if (source_ops.contains(user)) { in SafeToMerge()
125 if (source_ops.contains(successor)) { in SafeToMerge()
/external/tensorflow/tensorflow/python/eager/
Dlift_to_graph.py312 source_ops = set()
317 source_ops.add(s.op)
326 source_ops.add(s.op)
338 if op in source_ops or op in op_map: