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

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. 

16 

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 

24 

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 

32 

33 

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