251070eb2d7dc8376c868b8a86342c3332e706f0 |
|
19-Oct-2017 |
Igor Ganichev <iga@google.com> |
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
/external/tensorflow/tensorflow/python/util/util.h
|