Home
last modified time | relevance | path

Searched refs:CUDNN_RNN_TANH (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
Dcudnn_rnn_test.py67 CUDNN_RNN_TANH = cudnn_rnn_ops.CUDNN_RNN_TANH variable
112 elif rnn_mode == CUDNN_RNN_TANH:
242 elif mode == CUDNN_RNN_TANH:
485 elif rnn_mode == CUDNN_RNN_TANH:
704 self._TestSaveRestoreHelper(CUDNN_RNN_TANH)
866 self._TestCudnnCompatibleRnnCellsHelper(CUDNN_RNN_TANH)
1070 rnns = [CUDNN_LSTM, CUDNN_GRU, CUDNN_RNN_RELU, CUDNN_RNN_TANH]
1429 self._TestSimpleTrainingHelper(CUDNN_RNN_TANH, test_configs)
1448 self._TestSimpleTrainingHelper(CUDNN_RNN_TANH, test_configs)
1467 self._TestSimpleTrainingHelper(CUDNN_RNN_TANH, test_configs)
Dcudnn_rnn_ops_test.py57 CUDNN_RNN_TANH = cudnn_rnn_ops.CUDNN_RNN_TANH variable
1089 elif rnn_mode == CUDNN_RNN_TANH:
1111 CUDNN_LSTM, CUDNN_GRU, CUDNN_RNN_RELU, CUDNN_RNN_TANH
1163 CUDNN_LSTM, CUDNN_GRU, CUDNN_RNN_RELU, CUDNN_RNN_TANH
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/
Dcudnn_rnn.py37 CUDNN_RNN_TANH = cudnn_rnn_ops.CUDNN_RNN_TANH variable
596 _rnn_mode = CUDNN_RNN_TANH
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/
Dcudnn_rnn_ops.py43 CUDNN_RNN_TANH = "rnn_tanh" variable
611 _rnn_mode = CUDNN_RNN_TANH
906 if rnn_mode not in (CUDNN_LSTM, CUDNN_GRU, CUDNN_RNN_TANH, CUDNN_RNN_RELU):
908 (rnn_mode, CUDNN_LSTM, CUDNN_GRU, CUDNN_RNN_TANH,
942 elif rnn_mode == CUDNN_RNN_TANH:
1311 CUDNN_RNN_TANH, sequence_lengths, time_major,
1885 _rnn_mode = CUDNN_RNN_TANH
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1189 case CUDNN_RNN_TANH: in Create()