Home
last modified time | relevance | path

Searched refs:CUDNN_RNN_BIDIRECTION (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/
Dcudnn_rnn.py33 CUDNN_RNN_BIDIRECTION = cudnn_rnn_ops.CUDNN_RNN_BIDIRECTION variable
462 is_bidi = self._direction == CUDNN_RNN_BIDIRECTION
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/
Dcudnn_rnn_ops.py39 CUDNN_RNN_BIDIRECTION = "bidirectional" variable
438 if self._direction == CUDNN_RNN_BIDIRECTION:
520 if self._direction == CUDNN_RNN_BIDIRECTION:
583 if self._direction == CUDNN_RNN_BIDIRECTION:
921 if direction not in (CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION):
923 (direction, CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION))
1813 if direction not in (CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION):
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
Dcudnn_rnn_ops_test.py52 CUDNN_RNN_BIDIRECTION = cudnn_rnn_ops.CUDNN_RNN_BIDIRECTION variable
990 cudnn_rnn_ops.CUDNN_RNN_BIDIRECTION)
1064 cudnn_rnn_ops.CUDNN_RNN_BIDIRECTION)
1113 "direction": [CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION]
1165 "direction": [CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION]
Dcudnn_rnn_test.py69 CUDNN_RNN_BIDIRECTION = cudnn_rnn_ops.CUDNN_RNN_BIDIRECTION variable
665 directions = [CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION]
904 directions = [CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION]
1068 directions = [CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION]
1215 if dir_count == 1 else CUDNN_RNN_BIDIRECTION)