Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
Dcudnn_rnn_test.py65 CUDNN_GRU = cudnn_rnn_ops.CUDNN_GRU variable
109 elif rnn_mode == CUDNN_GRU:
240 elif mode == CUDNN_GRU:
483 elif rnn_mode == CUDNN_GRU:
699 self._TestSaveRestoreHelper(CUDNN_GRU)
861 self._TestCudnnCompatibleRnnCellsHelper(CUDNN_GRU)
1070 rnns = [CUDNN_LSTM, CUDNN_GRU, CUDNN_RNN_RELU, CUDNN_RNN_TANH]
1373 self._TestSimpleTrainingHelper(CUDNN_GRU, test_configs)
1392 self._TestSimpleTrainingHelper(CUDNN_GRU, test_configs)
1411 self._TestSimpleTrainingHelper(CUDNN_GRU, test_configs)
Dcudnn_rnn_ops_test.py55 CUDNN_GRU = cudnn_rnn_ops.CUDNN_GRU variable
654 rnn_mode=cudnn_rnn_ops.CUDNN_GRU)
1021 cudnn_rnn_ops.CUDNN_GRU,
1087 elif rnn_mode == CUDNN_GRU:
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.py35 CUDNN_GRU = cudnn_rnn_ops.CUDNN_GRU variable
589 _rnn_mode = CUDNN_GRU
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/
Dcudnn_rnn_ops.py41 CUDNN_GRU = "gru" variable
496 _rnn_mode = CUDNN_GRU
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,
938 elif rnn_mode == CUDNN_GRU:
1205 return _cudnn_rnn_no_input_c(inputs, input_h, params, is_training, CUDNN_GRU,
1876 _rnn_mode = CUDNN_GRU
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1193 case CUDNN_GRU: in Create()