code: def _ragged_batch(
      input_dataset,
      batch_size,
      drop_remainder=False,
      row_splits_dtype=dtypes.int64,
      name=None,
  ):
      ragged_dataset = _DenseToRaggedDataset(
          input_dataset, row_splits_dtype, name
      )
      return ragged_dataset.batch(batch_size, drop_remainder)
  
file_path: tensorflow-upstream/tensorflow/python/data/ops/ragged_batch_op.py
imports: from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.data.ops import structured_function
  from tensorflow.python.data.util import nest
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import tensor
  from tensorflow.python.ops.ragged import ragged_tensor
label_desc: Use image node_id 1 for calling a global function with example usage: _ragged_batch(input_dataset, batch_size, drop_remainder, row_splits_dtype, name) and returns: ragged_dataset