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

internal cleanups

Browse Source 
PiperOrigin-RevId: 235629187
This commit is contained in:
A. Unique TensorFlower 2019-02-25 17:02:45 -08:00 committed by TensorFlower Gardener
parent 0956cbe9de
commit 5bce34fe7d
3 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/python/op_hint.py
View File

@ -1140,7 +1140,6 @@ def _get_correct_mapping(original_index, nodes):
return original_index return original_index
@_tf_export("lite.convert_op_hints_to_stubs")
def _convert_op_hints_to_stubs_helper( def _convert_op_hints_to_stubs_helper(
graph_def, write_callback=lambda sess, graph_def: None): graph_def, write_callback=lambda sess, graph_def: None):
"""Converts a graph_def to a new graph_def where all op hints are stubbed. """Converts a graph_def to a new graph_def where all op hints are stubbed.
@ -1248,6 +1247,7 @@ def find_all_hinted_output_nodes(session=None, graph_def=None):
return hinted_outputs_nodes return hinted_outputs_nodes
@_tf_export("lite.experimental.convert_op_hints_to_stubs")
def convert_op_hints_to_stubs(session=None, def convert_op_hints_to_stubs(session=None,
graph_def=None, graph_def=None,
write_callback=lambda graph_def, comments: None): write_callback=lambda graph_def, comments: None):

4
tensorflow/tools/api/golden/v1/tensorflow.lite.experimental.pbtxt
View File

@ -4,4 +4,8 @@ tf_module {
name: "nn" name: "nn"
mtype: "<type \'module\'>" mtype: "<type \'module\'>"
} }
member_method {
name: "convert_op_hints_to_stubs"
argspec: "args=[\'session\', \'graph_def\', \'write_callback\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'<function <lambda> instance>\'], "
}
} }

4
tensorflow/tools/api/golden/v2/tensorflow.lite.experimental.pbtxt
View File

@ -4,4 +4,8 @@ tf_module {
name: "nn" name: "nn"
mtype: "<type \'module\'>" mtype: "<type \'module\'>"
} }
member_method {
name: "convert_op_hints_to_stubs"
argspec: "args=[\'session\', \'graph_def\', \'write_callback\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'<function <lambda> instance>\'], "
}
} }