diff --git a/tensorflow/python/autograph/impl/api.py b/tensorflow/python/autograph/impl/api.py index c8e96797a6d..cdc6d71ea1b 100644 --- a/tensorflow/python/autograph/impl/api.py +++ b/tensorflow/python/autograph/impl/api.py @@ -60,7 +60,7 @@ def is_autograph_strict_conversion_mode(): # to write converter. def convert( recursive=False, - optional_features=converter.Feature.ALL): + optional_features=None): """Decorator that compiles a function to use TensorFlow ops. The decorator is dynamic - it recompiles the target whenever the decorated @@ -370,7 +370,7 @@ def to_graph(entity, recursive=True, arg_values=None, arg_types=None, - experimental_optional_features=converter.Feature.ALL): + experimental_optional_features=None): """Converts a Python entity into a TensorFlow graph. Also see: `tf.autograph.to_code`, `tf.function`. @@ -454,7 +454,7 @@ def to_code(entity, arg_values=None, arg_types=None, indentation=' ', - experimental_optional_features=converter.Feature.ALL): + experimental_optional_features=None): """Similar to `to_graph`, but returns Python source code as a string. Also see: `tf.autograph.to_graph`. diff --git a/tensorflow/tools/api/golden/v1/tensorflow.autograph.pbtxt b/tensorflow/tools/api/golden/v1/tensorflow.autograph.pbtxt index 8880ed4f0cb..37351f1ccf4 100644 --- a/tensorflow/tools/api/golden/v1/tensorflow.autograph.pbtxt +++ b/tensorflow/tools/api/golden/v1/tensorflow.autograph.pbtxt @@ -10,11 +10,11 @@ tf_module { } member_method { 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 { 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 { name: "trace" diff --git a/tensorflow/tools/api/golden/v2/tensorflow.autograph.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.autograph.pbtxt index 8880ed4f0cb..37351f1ccf4 100644 --- a/tensorflow/tools/api/golden/v2/tensorflow.autograph.pbtxt +++ b/tensorflow/tools/api/golden/v2/tensorflow.autograph.pbtxt @@ -10,11 +10,11 @@ tf_module { } member_method { 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 { 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 { name: "trace"