1# Copyright 2016 The TensorFlow Authors. All Rights Reserved.
2#
3# Licensed under the Apache License, Version 2.0 (the "License");
4# you may not use this file except in compliance with the License.
5# You may obtain a copy of the License at
6#
7#     http://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS,
11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12# See the License for the specific language governing permissions and
13# limitations under the License.
14# ==============================================================================
15"""RNN Cells and additional RNN operations.
16
17See @{$python/contrib.rnn} guide.
18
19<!--From core-->
20@@RNNCell
21@@LayerRNNCell
22@@BasicRNNCell
23@@BasicLSTMCell
24@@GRUCell
25@@LSTMCell
26@@LSTMStateTuple
27@@DropoutWrapper
28@@MultiRNNCell
29@@DeviceWrapper
30@@ResidualWrapper
31
32<!--Used to be in core, but kept in contrib.-->
33@@EmbeddingWrapper
34@@InputProjectionWrapper
35@@OutputProjectionWrapper
36
37<!--Created in contrib, eventual plans to move to core.-->
38@@LayerNormBasicLSTMCell
39@@LSTMBlockWrapper
40@@LSTMBlockCell
41@@GRUBlockCell
42@@GRUBlockCellV2
43@@FusedRNNCell
44@@FusedRNNCellAdaptor
45@@TimeReversedFusedRNN
46@@LSTMBlockFusedCell
47@@CoupledInputForgetGateLSTMCell
48@@TimeFreqLSTMCell
49@@GridLSTMCell
50@@BidirectionalGridLSTMCell
51@@NASCell
52@@UGRNNCell
53@@IntersectionRNNCell
54@@PhasedLSTMCell
55@@ConvLSTMCell
56@@Conv1DLSTMCell
57@@Conv2DLSTMCell
58@@Conv3DLSTMCell
59@@HighwayWrapper
60@@GLSTMCell
61
62<!--RNNCell wrappers-->
63@@AttentionCellWrapper
64@@CompiledWrapper
65
66<!--RNN functions-->
67@@static_rnn
68@@static_state_saving_rnn
69@@static_bidirectional_rnn
70@@stack_bidirectional_dynamic_rnn
71@@stack_bidirectional_rnn
72
73<!--RNN utilities-->
74@@transpose_batch_time
75@@best_effort_input_batch_size
76"""
77
78from __future__ import absolute_import
79from __future__ import division
80from __future__ import print_function
81
82# pylint: disable=unused-import,wildcard-import,line-too-long
83from tensorflow.contrib.rnn.python.ops.core_rnn_cell import EmbeddingWrapper
84from tensorflow.contrib.rnn.python.ops.core_rnn_cell import InputProjectionWrapper
85from tensorflow.contrib.rnn.python.ops.core_rnn_cell import OutputProjectionWrapper
86
87from tensorflow.contrib.rnn.python.ops.fused_rnn_cell import *
88from tensorflow.contrib.rnn.python.ops.gru_ops import *
89from tensorflow.contrib.rnn.python.ops.lstm_ops import *
90from tensorflow.contrib.rnn.python.ops.rnn import *
91from tensorflow.contrib.rnn.python.ops.rnn_cell import *
92
93from tensorflow.python.ops.rnn import _best_effort_input_batch_size as best_effort_input_batch_size
94from tensorflow.python.ops.rnn import _transpose_batch_time as transpose_batch_time
95from tensorflow.python.ops.rnn import static_bidirectional_rnn
96from tensorflow.python.ops.rnn import static_rnn
97from tensorflow.python.ops.rnn import static_state_saving_rnn
98
99from tensorflow.python.ops.rnn_cell import *
100# pylint: enable=unused-import,wildcard-import,line-too-long
101
102from tensorflow.python.util.all_util import remove_undocumented
103remove_undocumented(__name__)
104