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

Switch default for autograph.(to_graph|to_code|convert) to optional_features=None to better align with tf.function

Browse Source 
PiperOrigin-RevId: 243108206
This commit is contained in:
Brian Lee 2019-04-11 11:30:38 -07:00 committed by TensorFlower Gardener
parent 1942fea1b8
commit 80a47e09c9
3 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/autograph/impl/api.py
View File

@ -60,7 +60,7 @@ def is_autograph_strict_conversion_mode():
# to write converter. # to write converter.
def convert( def convert(
recursive=False, recursive=False,
optional_features=converter.Feature.ALL): optional_features=None):
"""Decorator that compiles a function to use TensorFlow ops. """Decorator that compiles a function to use TensorFlow ops.
The decorator is dynamic - it recompiles the target whenever the decorated The decorator is dynamic - it recompiles the target whenever the decorated
@ -370,7 +370,7 @@ def to_graph(entity,
recursive=True, recursive=True,
arg_values=None, arg_values=None,
arg_types=None, arg_types=None,
experimental_optional_features=converter.Feature.ALL): experimental_optional_features=None):
"""Converts a Python entity into a TensorFlow graph. """Converts a Python entity into a TensorFlow graph.
Also see: `tf.autograph.to_code`, `tf.function`. Also see: `tf.autograph.to_code`, `tf.function`.
@ -454,7 +454,7 @@ def to_code(entity,
arg_values=None, arg_values=None,
arg_types=None, arg_types=None,
indentation=' ', indentation=' ',
experimental_optional_features=converter.Feature.ALL): experimental_optional_features=None):
"""Similar to `to_graph`, but returns Python source code as a string. """Similar to `to_graph`, but returns Python source code as a string.
Also see: `tf.autograph.to_graph`. Also see: `tf.autograph.to_graph`.

4
tensorflow/tools/api/golden/v1/tensorflow.autograph.pbtxt
View File

@ -10,11 +10,11 @@ tf_module {
} }
member_method { member_method {
name: "to_code" name: "to_code"
argspec: "args=[\'entity\', \'recursive\', \'arg_values\', \'arg_types\', \'indentation\', \'experimental_optional_features\'], varargs=None, keywords=None, defaults=[\'True\', \'None\', \'None\', \' \', \'Feature.ALL\'], " argspec: "args=[\'entity\', \'recursive\', \'arg_values\', \'arg_types\', \'indentation\', \'experimental_optional_features\'], varargs=None, keywords=None, defaults=[\'True\', \'None\', \'None\', \' \', \'None\'], "
} }
member_method { member_method {
name: "to_graph" name: "to_graph"
argspec: "args=[\'entity\', \'recursive\', \'arg_values\', \'arg_types\', \'experimental_optional_features\'], varargs=None, keywords=None, defaults=[\'True\', \'None\', \'None\', \'Feature.ALL\'], " argspec: "args=[\'entity\', \'recursive\', \'arg_values\', \'arg_types\', \'experimental_optional_features\'], varargs=None, keywords=None, defaults=[\'True\', \'None\', \'None\', \'None\'], "
} }
member_method { member_method {
name: "trace" name: "trace"

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

@ -10,11 +10,11 @@ tf_module {
} }
member_method { member_method {
name: "to_code" name: "to_code"
argspec: "args=[\'entity\', \'recursive\', \'arg_values\', \'arg_types\', \'indentation\', \'experimental_optional_features\'], varargs=None, keywords=None, defaults=[\'True\', \'None\', \'None\', \' \', \'Feature.ALL\'], " argspec: "args=[\'entity\', \'recursive\', \'arg_values\', \'arg_types\', \'indentation\', \'experimental_optional_features\'], varargs=None, keywords=None, defaults=[\'True\', \'None\', \'None\', \' \', \'None\'], "
} }
member_method { member_method {
name: "to_graph" name: "to_graph"
argspec: "args=[\'entity\', \'recursive\', \'arg_values\', \'arg_types\', \'experimental_optional_features\'], varargs=None, keywords=None, defaults=[\'True\', \'None\', \'None\', \'Feature.ALL\'], " argspec: "args=[\'entity\', \'recursive\', \'arg_values\', \'arg_types\', \'experimental_optional_features\'], varargs=None, keywords=None, defaults=[\'True\', \'None\', \'None\', \'None\'], "
} }
member_method { member_method {
name: "trace" name: "trace"