if (!value_type) return failure();
    auto feature_dim = GetFeatureDimension(data_format, value_type);
    auto bias_broadcast = Broadcast1DToFeatureDim(
        loc, op.getValue(), op.getBias(), feature_dim, rewriter);
    Value add = rewriter.create<AddOp>(loc, op.getValue(), bias_broadcast);
    if (add.getType() != op.getType()) {
      add = rewriter.create<tensor::CastOp>(loc, op.getType(), add);
    }