code: def _take_while(
      input_dataset, predicate, name=None
  ):  # pylint: disable=unused-private-name
      """See `Dataset.take_while()` for details."""
      return _TakeWhileDataset(input_dataset, predicate, name=name)
  
file_path: tensorflow-upstream/tensorflow/python/data/ops/take_while_op.py
imports: from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.data.ops import structured_function
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import tensor_spec
  from tensorflow.python.ops import gen_experimental_dataset_ops
label_desc: Use image node_id 1 for calling a global function with example usage: _take_while(input_dataset, predicate, name) and returns: _TakeWhileDataset