Searched defs:IfRegionOp (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | merge_control_flow.cc | 191 const llvm::SmallVector<mlir::TF::IfRegionOp, 8>& if_ops, in CanAddToIfSegment()
230 const llvm::SmallVector<mlir::TF::IfRegionOp, 8>& if_ops, in FindLastIfInSegment()
290 llvm::SmallVector<TF::IfRegionOp, 8>& sub_if_group, in GetMoveOpList()
327 const llvm::SmallVector<mlir::TF::IfRegionOp, 8>& if_ops, in GenerateSegmentsPerIfGroups()
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()
[all …]
|