class_docstr: Defines a toy module.
class_docstr_tok: ["Defines","a","toy","module","."]
code: def map_data(self, x):
      dataset = dataset_ops.Dataset.from_tensors(x)
      r = dataset.map(lambda x: x + 1)
      return r.get_single_element()
  
file_path: tensorflow-upstream/tensorflow/core/tfrt/saved_model/tests/gen_data.py
imports: from absl import app
  from absl import flags
  from absl import logging
  from tensorflow.python.compat import v2_compat
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import tensor_spec
  from tensorflow.python.module import module
  from tensorflow.python.saved_model import save
  from tensorflow.python.saved_model import save_options
label_desc: Use image node_id 1 for calling the ToyModule obj's underlying member method code with example usage: obj.map_data(x) and returns: r