From 80a47e09c97f8b7d4b45d3ed86c09b20ffca3ea4 Mon Sep 17 00:00:00 2001 From: Brian Lee Date: Thu, 11 Apr 2019 11:30:38 -0700 Subject: [PATCH] Switch default for autograph.(to_graph|to_code|convert) to optional_features=None to better align with tf.function PiperOrigin-RevId: 243108206 --- tensorflow/python/autograph/impl/api.py | 6 +++--- tensorflow/tools/api/golden/v1/tensorflow.autograph.pbtxt | 4 ++-- tensorflow/tools/api/golden/v2/tensorflow.autograph.pbtxt | 4 ++-- 3 files changed, 7 insertions(+), 7 deletions(-) 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"