code: def _c4_dataset(split="train"):
      return _load_dataset("c4:2.3.0", split=split)
  
imports: import collections
  import os
  from unittest import mock
  import gin
  import numpy
  from t5.data import assert_dataset
  from t5.data import preprocessors
  import tensorflow
  import tensorflow_datasets
  from trax.data import inputs
  from trax.data import tf_inputs
label_desc: Use image node_id 3 for calling a global function with example usage: _c4_dataset(split) and returns: _load_dataset