diff options
Diffstat (limited to 'tensorflow/contrib/autograph/__init__.py')
-rw-r--r-- | tensorflow/contrib/autograph/__init__.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/contrib/autograph/__init__.py b/tensorflow/contrib/autograph/__init__.py index 310eb34a70..637e49c082 100644 --- a/tensorflow/contrib/autograph/__init__.py +++ b/tensorflow/contrib/autograph/__init__.py @@ -29,6 +29,8 @@ from tensorflow.contrib.autograph.impl.api import do_not_convert from tensorflow.contrib.autograph.impl.api import RunMode from tensorflow.contrib.autograph.impl.api import to_code from tensorflow.contrib.autograph.impl.api import to_graph +from tensorflow.contrib.autograph.impl.directives import set_element_type +from tensorflow.contrib.autograph.impl.directives import set_loop_options from tensorflow.contrib.autograph.impl.special_functions import stack from tensorflow.contrib.autograph.pyct.transformer import AutographParseError from tensorflow.python.util.all_util import remove_undocumented @@ -41,7 +43,9 @@ _allowed_symbols = [ 'do_not_convert', 'to_code', 'to_graph', - # Special functions + # Special functions and directives + 'set_element_type', + 'set_loop_options', 'stack', # Exceptions 'AutographParseError', |