Searched refs:infeed_op (Results 1 – 3 of 3) sorted by relevance
15 "infeed_op.cc",
1112 auto infeed_op = CreateOpWithoutAttrs<lmhlo::InfeedOp>(instr, operands); in EmitInfeedOp() local1113 infeed_op.configAttr(builder_.getStringAttr(infeed->infeed_config())); in EmitInfeedOp()1114 return infeed_op; in EmitInfeedOp()
3193 auto infeed_op = mlir::cast<mlir::lmhlo::InfeedOp>(input.op); in HandleInfeed() local3196 dest_slices.reserve(infeed_op.outputs().size()); in HandleInfeed()3198 for (mlir::Value output : infeed_op.outputs()) { in HandleInfeed()