Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/tensorflow/python/ops/rnn_cell_impl.py: 100%
19 statements
« prev ^ index » next coverage.py v7.4.0, created at 2024-01-03 07:57 +0000
« prev ^ index » next coverage.py v7.4.0, created at 2024-01-03 07:57 +0000
1# Copyright 2015 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"""Module implementing RNN Cells.
17This module provides a number of basic commonly used RNN cells, such as LSTM
18(Long Short Term Memory) or GRU (Gated Recurrent Unit), and a number of
19operators that allow adding dropouts, projections, or embeddings for inputs.
20Constructing multi-layer cells is supported by the class `MultiRNNCell`, or by
21calling the `rnn` ops several times.
22"""
23from tensorflow.python.keras.layers.legacy_rnn import rnn_cell_impl
25# Remove caller that rely on private symbol in future.
26# pylint: disable=protected-access
27_BIAS_VARIABLE_NAME = rnn_cell_impl._BIAS_VARIABLE_NAME
28_WEIGHTS_VARIABLE_NAME = rnn_cell_impl._WEIGHTS_VARIABLE_NAME
29_concat = rnn_cell_impl._concat
30_zero_state_tensors = rnn_cell_impl._zero_state_tensors
31# pylint: disable=protected-access
34assert_like_rnncell = rnn_cell_impl.assert_like_rnncell
35ASSERT_LIKE_RNNCELL_ERROR_REGEXP = rnn_cell_impl.ASSERT_LIKE_RNNCELL_ERROR_REGEXP # pylint: disable=line-too-long
36BasicLSTMCell = rnn_cell_impl.BasicLSTMCell
37BasicRNNCell = rnn_cell_impl.BasicRNNCell
38DeviceWrapper = rnn_cell_impl.DeviceWrapper
39DropoutWrapper = rnn_cell_impl.DropoutWrapper
40GRUCell = rnn_cell_impl.GRUCell
41LayerRNNCell = rnn_cell_impl.LayerRNNCell
42LSTMCell = rnn_cell_impl.LSTMCell
43LSTMStateTuple = rnn_cell_impl.LSTMStateTuple
44MultiRNNCell = rnn_cell_impl.MultiRNNCell
45ResidualWrapper = rnn_cell_impl.ResidualWrapper
46RNNCell = rnn_cell_impl.RNNCell