experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Remove confusing unused output_gen field.

Browse Source 
PiperOrigin-RevId: 244119314
This commit is contained in:
Suharsh Sivakumar 2019-04-17 20:36:54 -07:00 committed by TensorFlower Gardener
parent ea73eb4a0c
commit c903fc2338
3 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/lite/python/lite.py
View File

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

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

@ -4,6 +4,6 @@ tf_class {
is_instance: "<type \'object\'>" is_instance: "<type \'object\'>"
member_method { member_method {
name: "__init__" 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"
} }
} }

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

@ -4,6 +4,6 @@ tf_class {
is_instance: "<type \'object\'>" is_instance: "<type \'object\'>"
member_method { member_method {
name: "__init__" 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"
} }
} }