diff options
| author |  Igor Ganichev <iga@google.com> | 2017-10-31 11:25:58 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-31 11:29:51 -0700 |
| commit | 8a09bbc4a5de2cb8db20dd41112abec245eaff88 (patch) | |
| tree | 67783676fe1617a1e29f60866e9d556fe71a80db /tensorflow/python/pywrap_tfe.i | |
| parent | 585432cc21f52ece2c7fd9bd21a45d40b1e63f42 (diff) | |
Add TFE_Py_TensorShapeSlice function
TFE_Py_TensorShapeSlice takes a list of EagerTensors and returns a list
of their i'th dimensions. This utility is fairly niche but it is simple
and reduces SPINN training time by over 12%.
PiperOrigin-RevId: 174065044
Diffstat (limited to 'tensorflow/python/pywrap_tfe.i')
| -rw-r--r-- | tensorflow/python/pywrap_tfe.i | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/pywrap_tfe.i b/tensorflow/python/pywrap_tfe.i index fa36b77311..637f738fed 100644 --- a/tensorflow/python/pywrap_tfe.i +++ b/tensorflow/python/pywrap_tfe.i @@ -34,6 +34,7 @@ limitations under the License. %rename("%s") TFE_ContextOptionsSetConfig; %rename("%s") TFE_ContextOptionsSetDevicePlacementPolicy; %rename("%s") TFE_DeleteContextOptions; +%rename("%s") TFE_Py_TensorShapeSlice; %{ #include "tensorflow/python/eager/pywrap_tfe.h" |