Remove confusing unused output_gen field.

PiperOrigin-RevId: 244119314

tensorflow/lite/python/lite.py

@@ -109,7 +109,7 @@ class RepresentativeDataset(object):
   converted floating point model.
   """
 
-  def __init__(self, input_gen, output_gen=None):
+  def __init__(self, input_gen):
     """Creates a representative dataset.
 
     Args:
@@ -117,14 +117,8 @@ class RepresentativeDataset(object):
         for the model. This must be a callable object that returns an object
         that supports the `iter()` protocol (e.g. a generator function). The
         elements generated must have same type and shape as inputs to the model.
-      output_gen: (optional) an output generator that can be used to generate
-        output samples for the model. This must be a callable object that
-        returns an object that supports the `iter()` protocol (e.g. a generator
-        function). The elements generated must have same type and shape as
-        outputs to the model. (default None)
     """
     self.input_gen = input_gen
-    self.output_gen = output_gen
 
 
 @_tf_export("lite.TargetSpec")

tensorflow/tools/api/golden/v1/tensorflow.lite.-representative-dataset.pbtxt

@@ -4,6 +4,6 @@ tf_class {
   is_instance: "<type \'object\'>"
   member_method {
     name: "__init__"
-    argspec: "args=[\'self\', \'input_gen\', \'output_gen\'], varargs=None, keywords=None, defaults=[\'None\'], "
+    argspec: "args=[\'self\', \'input_gen\'], varargs=None, keywords=None, defaults=None"
   }
 }

tensorflow/tools/api/golden/v2/tensorflow.lite.-representative-dataset.pbtxt

@@ -4,6 +4,6 @@ tf_class {
   is_instance: "<type \'object\'>"
   member_method {
     name: "__init__"
-    argspec: "args=[\'self\', \'input_gen\', \'output_gen\'], varargs=None, keywords=None, defaults=[\'None\'], "
+    argspec: "args=[\'self\', \'input_gen\'], varargs=None, keywords=None, defaults=None"
   }
 }