Searched refs:GetYield (Results 1 – 9 of 9) sorted by relevance
100 for (Value operand : island_op.GetYield().getOperands()) in runOnOperation()116 YieldOp yield_op = island_op.GetYield(); in runOnOperation()
146 llvm::zip(parent.GetYield().getOperands(), parent.outputs())) { in GetNewIslandResultsAndForwardResults()162 llvm::zip(child.GetYield().getOperands(), child.outputs())) { in GetNewIslandResultsAndForwardResults()
237 YieldOp yield_op = island.GetYield(); in MergeIsland()
401 ValuePort(island.GetYield().fetches()[port[0]]), values); in ComputeOutputComponent()936 island_op.GetYield(), island_op.GetYield().fetches(), op->getResults()); in InferShapeForNonTFDialectOperation()
131 tf_executor::YieldOp yield = island.GetYield(); in PopulateEmptyIsland()229 for (auto fetch : island_op.GetYield().fetches()) { in BreakUpIsland()320 llvm::zip(island_op.outputs(), island_op.GetYield().fetches())) in BreakUpIsland()
59 llvm::zip(island_op.outputs(), island_op.GetYield().fetches())) in LiftIslandOpInnerOpsFromGraph()
262 YieldOp IslandOp::GetYield() { return llvm::cast<YieldOp>(GetBody().back()); } in GetYield() function in mlir::tf_executor::IslandOp271 YieldOp yield = GetYield(); in WrapsSingleOp()326 YieldOp yield_op = op.GetYield(); in Print()987 YieldOp yield_op = island_op.GetYield(); in matchAndRewrite()1065 rewriter.replaceOp(op, {op.GetYield().getOperand(0), nullptr}); in matchAndRewrite()
211 YieldOp GetYield();
193 value = island.GetYield().getOperand(res_index); in BacktrackValue()