diff options
author | 2018-03-22 14:53:59 -0700 | |
---|---|---|
committer | 2018-03-22 14:56:38 -0700 | |
commit | 63d46266ba5b2a513244e13321f76e7acd03aba3 (patch) | |
tree | cf1b4e9dde164e07c219674a711edb1cae68b36e /tensorflow/python/__init__.py | |
parent | 730e69519a93a668d97ea298d52365326c00357d (diff) |
Move cuDNN RNN ops to core, for use in the internal TF codebase only (not publicly exposed).
RELNOTES: Moved cuDNN RNN ops to core.
PiperOrigin-RevId: 190130405
Diffstat (limited to 'tensorflow/python/__init__.py')
-rw-r--r-- | tensorflow/python/__init__.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/python/__init__.py b/tensorflow/python/__init__.py index 5a9cd7531d..3346937904 100644 --- a/tensorflow/python/__init__.py +++ b/tensorflow/python/__init__.py @@ -99,6 +99,10 @@ from tensorflow.python.user_ops import user_ops from tensorflow.python.util import compat +# Import cudnn rnn ops to make sure their ops are registered. +from tensorflow.python.ops import gen_cudnn_rnn_ops as _ + + # Import the names from python/training.py as train.Name. from tensorflow.python.training import training as train |