Home
last modified time | relevance | path

Searched refs:_rnn_mode (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/
Dcudnn_rnn.py151 _rnn_mode = None variable in _CudnnRNN
259 return self._rnn_mode
423 if self._rnn_mode == CUDNN_LSTM:
428 if self._rnn_mode == CUDNN_LSTM:
435 if self._rnn_mode == CUDNN_LSTM:
487 rnn_mode=self._rnn_mode,
506 self._rnn_mode,
543 _rnn_mode = CUDNN_LSTM variable in CudnnLSTM
589 _rnn_mode = CUDNN_GRU variable in CudnnGRU
596 _rnn_mode = CUDNN_RNN_TANH variable in CudnnRNNTanh
[all …]
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/
Dcudnn_rnn_ops.py244 rnn_mode=self._rnn_mode,
265 rnn_mode=self._rnn_mode,
405 _rnn_mode = CUDNN_LSTM variable in CudnnParamsFormatConverterLSTM
496 _rnn_mode = CUDNN_GRU variable in CudnnParamsFormatConverterGRU
611 _rnn_mode = CUDNN_RNN_TANH variable in CudnnParamsFormatConverterTanh
617 _rnn_mode = CUDNN_RNN_RELU variable in CudnnParamsFormatConverterRelu
1542 self._rnn_mode = rnn_mode
1567 return self._rnn_mode
1580 rnn_mode=self._rnn_mode,
1633 self._rnn_mode,
[all …]