aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/__init__.py
diff options
context:
space:
mode:
authorGravatar Yao Zhang <yaozhang@google.com>2017-03-22 16:02:45 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-03-22 17:26:30 -0700
commit9bca1d4e10d8e50140f1b5c735a761065d553f24 (patch)
treebb32152f3234eec71bead0f6c18c0d24de00b381 /tensorflow/python/__init__.py
parentd6e3cdd4e6eb7982d5ee7cf5845ccff168712f8e (diff)
Add RewriterConfig the to python API.
Change: 150948079
Diffstat (limited to 'tensorflow/python/__init__.py')
-rw-r--r--tensorflow/python/__init__.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/__init__.py b/tensorflow/python/__init__.py
index e632091094..3663d8af7a 100644
--- a/tensorflow/python/__init__.py
+++ b/tensorflow/python/__init__.py
@@ -57,6 +57,7 @@ from tensorflow.core.framework.summary_pb2 import *
from tensorflow.core.framework.attr_value_pb2 import *
from tensorflow.core.protobuf.meta_graph_pb2 import TensorInfo
from tensorflow.core.protobuf.config_pb2 import *
+from tensorflow.core.protobuf.rewriter_config_pb2 import *
from tensorflow.core.util.event_pb2 import *
# Framework
@@ -145,6 +146,7 @@ _allowed_symbols = [
'NameAttrList',
'NodeDef',
'OptimizerOptions',
+ 'RewriterConfig',
'RunOptions',
'RunMetadata',
'SessionLog',