Searched refs:outfeed_op (Results 1 – 2 of 2) sorted by relevance
1126 auto outfeed_op = CreateOpWithoutAttrs<lmhlo::OutfeedOp>(instr, operands); in EmitOutfeedOp() local1127 outfeed_op.configAttr(builder_.getStringAttr(outfeed->outfeed_config())); in EmitOutfeedOp()1128 return outfeed_op; in EmitOutfeedOp()
3212 auto outfeed_op = mlir::cast<mlir::lmhlo::OutfeedOp>(input.op); in HandleOutfeed() local3215 source_slices.reserve(outfeed_op.operands().size()); in HandleOutfeed()3217 for (mlir::Value operand : outfeed_op.operands()) { in HandleOutfeed()