Home
last modified time | relevance | path

Searched defs:while_ops (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddtensor_allreduce_sum_optimization.cc445 const llvm::SmallVectorImpl<mlir::TF::WhileRegionOp>& while_ops, in ApplyOptimization()
471 llvm::SmallVectorImpl<mlir::TF::WhileRegionOp>* while_ops) { in CollectOptimizationCandidates()
496 llvm::SmallVector<mlir::TF::WhileRegionOp, 4> while_ops; in runOnOperation() local
H A Dlayout_propagation_v2.cc1139 const llvm::SmallVectorImpl<mlir::TF::WhileRegionOp>& while_ops, in InsertRelayoutForWhileLoops()
1566 llvm::SmallVector<mlir::TF::WhileRegionOp, 4> while_ops; in runOnOperation() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dwhile_loop_simplifier.cc1349 std::vector<HloInstruction*> while_ops; in Run() local