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

Export lift_to_graph as a tf.__internal__ API.

Browse Source 
PiperOrigin-RevId: 353725510
Change-Id: I7021e1bcbb825c34bedd692bc00830802a0cfc2b
This commit is contained in:
Yanhui Liang 2021-01-25 14:02:43 -08:00 committed by TensorFlower Gardener
parent 1cc6050ebf
commit ba088e0369
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/eager/lift_to_graph.py
View File

@ -28,6 +28,7 @@ from tensorflow.python.ops import op_selector
from tensorflow.python.ops import resource_variable_ops
from tensorflow.python.util import compat
from tensorflow.python.util import object_identity
from tensorflow.python.util.tf_export import tf_export
UnliftableError = op_selector.UnliftableError
@ -202,6 +203,7 @@ def _copy_source(s, graph, op_map, handle_captures, inverse_captures,
op_map[s.op] = copied_placeholder.op
@tf_export("__internal__.lift_to_graph", v1=[])
def lift_to_graph(tensors,
graph,
sources=None,

4
tensorflow/tools/api/golden/v2/tensorflow.__internal__.pbtxt
View File

@ -76,4 +76,8 @@ tf_module {
name: "is_tfrt_enabled"
argspec: "args=[], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "lift_to_graph"
argspec: "args=[\'tensors\', \'graph\', \'sources\', \'disallowed_placeholders\', \'add_sources\', \'handle_captures\', \'base_graph\', \'op_map\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'False\', \'False\', \'None\', \'None\'], "
}
}