diff options
| author |  James Qin <jamesqin@google.com> | 2017-10-25 21:38:45 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-25 21:43:35 -0700 |
| commit | 62f62bfe9f0ffba8ae406dcb1b6c0165d38fe633 (patch) | |
| tree | cf8ee3da627a0378bce16814ad49318a0fe4cc41 /tensorflow/contrib/cudnn_rnn | |
| parent | 65616777d73913346ec446df80b6d7aa64e54b24 (diff) | |
Switch tf.contrib.cudnn_rnn.CudnnXXX to point to layer APIs instead of op wrappers
PiperOrigin-RevId: 173494053
Diffstat (limited to 'tensorflow/contrib/cudnn_rnn')
| -rw-r--r-- | tensorflow/contrib/cudnn_rnn/__init__.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/contrib/cudnn_rnn/__init__.py b/tensorflow/contrib/cudnn_rnn/__init__.py index 87ba834770..bc44562b50 100644 --- a/tensorflow/contrib/cudnn_rnn/__init__.py +++ b/tensorflow/contrib/cudnn_rnn/__init__.py @@ -30,13 +30,14 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function +from tensorflow.contrib.cudnn_rnn.python.layers.cudnn_rnn import CudnnGRU +from tensorflow.contrib.cudnn_rnn.python.layers.cudnn_rnn import CudnnLSTM +from tensorflow.contrib.cudnn_rnn.python.layers.cudnn_rnn import CudnnRNNRelu +from tensorflow.contrib.cudnn_rnn.python.layers.cudnn_rnn import CudnnRNNTanh from tensorflow.contrib.cudnn_rnn.python.ops.cudnn_rnn_ops import CudnnCompatibleGRUCell from tensorflow.contrib.cudnn_rnn.python.ops.cudnn_rnn_ops import CudnnCompatibleLSTMCell -from tensorflow.contrib.cudnn_rnn.python.ops.cudnn_rnn_ops import CudnnGRU from tensorflow.contrib.cudnn_rnn.python.ops.cudnn_rnn_ops import CudnnGRUSaveable -from tensorflow.contrib.cudnn_rnn.python.ops.cudnn_rnn_ops import CudnnLSTM from tensorflow.contrib.cudnn_rnn.python.ops.cudnn_rnn_ops import CudnnLSTMSaveable -from tensorflow.contrib.cudnn_rnn.python.ops.cudnn_rnn_ops import CudnnRNNRelu from tensorflow.contrib.cudnn_rnn.python.ops.cudnn_rnn_ops import CudnnRNNReluSaveable from tensorflow.contrib.cudnn_rnn.python.ops.cudnn_rnn_ops import CudnnRNNTanhSaveable |