diff options
author | Igor Ganichev <iga@google.com> | 2017-10-18 15:17:14 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-18 15:20:43 -0700 |
commit | 251070eb2d7dc8376c868b8a86342c3332e706f0 (patch) | |
tree | 5f674660ae5e0802e26fcf0e0cffb871d0d21842 /tensorflow/python/tensorflow.i | |
parent | 818644d0937d1e4b097b15d8c823835baba9fbc7 (diff) |
Implement nest.flatten in C++
This function is used extensively in imperative_grad. Implementing
it in C++ reduces SPINN training time by over 8%.
PiperOrigin-RevId: 172663591
Diffstat (limited to 'tensorflow/python/tensorflow.i')
-rw-r--r-- | tensorflow/python/tensorflow.i | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/tensorflow.i b/tensorflow/python/tensorflow.i index 9cef765bf3..d221dd523b 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/util/util.i" + %include "tensorflow/python/grappler/cluster.i" %include "tensorflow/python/grappler/item.i" %include "tensorflow/python/grappler/tf_optimizer.i" |