fill->setAttr(kICIWeightDistributionMlirBridgeMarker,
                builder.getBoolAttr(true));
  return fill;
}

// Add parallel collection of inputs to the replicated op.
LogicalResult AppendReplicatedInput(OpBuilder& builder, ReplicateOp replicate,
                                    ValueRange inputs, Type type,
                                    BlockArgument& block_arg) {