aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/tensorflow.i
diff options
context:
space:
mode:
authorGravatar Benoit Steiner <bsteiner@google.com>2017-09-27 17:54:39 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-09-27 17:58:52 -0700
commitcf07600653c01675fe339d604f42000074d9a976 (patch)
tree8c08fd00913713cd3d3a4b6b78793a0f9a423fe8 /tensorflow/python/tensorflow.i
parent6dc4aac4744876873c74c30678502c773a6318ca (diff)
Added a python API on top of Grappler items and Grappler clusters
PiperOrigin-RevId: 170280771
Diffstat (limited to 'tensorflow/python/tensorflow.i')
-rw-r--r--tensorflow/python/tensorflow.i2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/tensorflow.i b/tensorflow/python/tensorflow.i
index f15854d240..9cef765bf3 100644
--- a/tensorflow/python/tensorflow.i
+++ b/tensorflow/python/tensorflow.i
@@ -44,6 +44,8 @@ limitations under the License.
%include "tensorflow/python/util/transform_graph.i"
+%include "tensorflow/python/grappler/cluster.i"
+%include "tensorflow/python/grappler/item.i"
%include "tensorflow/python/grappler/tf_optimizer.i"
%include "tensorflow/python/grappler/cost_analyzer.i"
%include "tensorflow/python/grappler/model_analyzer.i"