Searched defs:if_op_segment (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | merge_control_flow.cc | 355 const llvm::SmallVector<TF::IfRegionOp, 8>& if_op_segment) { in GetReturnIndicesToKeep()
379 const llvm::SmallVector<TF::IfRegionOp, 8>& if_op_segment) { in GetReturnIndicesVec()
391 void ReplaceInternalUsage(llvm::SmallVector<TF::IfRegionOp, 8>& if_op_segment) { in ReplaceInternalUsage()
436 llvm::SmallVector<TF::IfRegionOp, 8>& if_op_segment, in ReplaceExternalUsage()
480 OpBuilder& builder, llvm::SmallVector<TF::IfRegionOp, 8>& if_op_segment, in CreateYieldOps()
519 llvm::SmallVector<TF::IfRegionOp, 8>& if_op_segment, in MergeIfPerSegment()
|