batchInputOp.getNumParallelCalls(), rewriter.getBoolAttr(false));

    if (op.getMetadata() != batchInputOp.getMetadata()) {
      return failure();
    }

    auto fused_op = rewriter.create<MapAndBatchDatasetOp>(
        op.getLoc(), op.getType(), batchInputOp.getInputDataset(),
        batchInputOp.getOtherArguments(), op.getBatchSize(),
        num_parallel_calls_op.getY(), op.getDropRemainder(),