Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/
Dcudnn_rnn_ops.py38 CUDNN_RNN_UNIDIRECTION = "unidirectional" variable
188 direction=CUDNN_RNN_UNIDIRECTION):
211 self._num_dirs = 1 if self._direction == CUDNN_RNN_UNIDIRECTION else 2
300 if self._direction == CUDNN_RNN_UNIDIRECTION:
371 if self._direction == CUDNN_RNN_UNIDIRECTION:
638 direction=CUDNN_RNN_UNIDIRECTION,
699 self._num_dirs = 1 if self._direction == CUDNN_RNN_UNIDIRECTION else 2
781 if self._direction == CUDNN_RNN_UNIDIRECTION:
796 if self._direction == CUDNN_RNN_UNIDIRECTION:
921 if direction not in (CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION):
[all …]
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/
Dcudnn_rnn.py32 CUDNN_RNN_UNIDIRECTION = cudnn_rnn_ops.CUDNN_RNN_UNIDIRECTION variable
161 direction=CUDNN_RNN_UNIDIRECTION,
268 return 1 if self._direction == CUDNN_RNN_UNIDIRECTION else 2
477 num_dirs = 1 if self._direction == CUDNN_RNN_UNIDIRECTION else 2
562 if self._direction == CUDNN_RNN_UNIDIRECTION:
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
Dcudnn_rnn_test.py68 CUDNN_RNN_UNIDIRECTION = cudnn_rnn_ops.CUDNN_RNN_UNIDIRECTION variable
89 direction=CUDNN_RNN_UNIDIRECTION,
278 direction = CUDNN_RNN_UNIDIRECTION
349 direction = CUDNN_RNN_UNIDIRECTION
378 direction = CUDNN_RNN_UNIDIRECTION
489 num_dirs = 1 if direction == CUDNN_RNN_UNIDIRECTION else 2
496 if direction == CUDNN_RNN_UNIDIRECTION:
665 directions = [CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION]
685 direction=CUDNN_RNN_UNIDIRECTION,
780 direction = CUDNN_RNN_UNIDIRECTION
[all …]
Dcudnn_rnn_ops_test.py51 CUDNN_RNN_UNIDIRECTION = cudnn_rnn_ops.CUDNN_RNN_UNIDIRECTION variable
931 num_dirs = 1 if direction == cudnn_rnn_ops.CUDNN_RNN_UNIDIRECTION else 2
979 cudnn_rnn_ops.CUDNN_RNN_UNIDIRECTION)
997 num_dirs = 1 if direction == cudnn_rnn_ops.CUDNN_RNN_UNIDIRECTION else 2
1053 cudnn_rnn_ops.CUDNN_RNN_UNIDIRECTION)
1113 "direction": [CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION]
1165 "direction": [CUDNN_RNN_UNIDIRECTION, CUDNN_RNN_BIDIRECTION]