Searched defs:input_mode (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| D | cudnn_rnn_ops.py | 195 input_mode=CUDNN_INPUT_LINEAR_MODE, argument
778 def check_input_mode(input_mode): argument
810 input_mode=CUDNN_INPUT_LINEAR_MODE, argument
869 input_mode=CUDNN_INPUT_LINEAR_MODE, argument
911 input_mode=CUDNN_INPUT_LINEAR_MODE, argument
954 input_mode=CUDNN_INPUT_LINEAR_MODE, argument
993 input_mode=CUDNN_INPUT_LINEAR_MODE, argument
1033 input_mode=CUDNN_INPUT_LINEAR_MODE, argument
1074 input_mode=CUDNN_INPUT_LINEAR_MODE, argument
1136 input_mode=CUDNN_INPUT_LINEAR_MODE, argument
[all …]
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/ |
| D | cudnn_rnn.py | 156 input_mode=CUDNN_INPUT_LINEAR_MODE, argument
225 def input_mode(self): member in _CudnnRNN
|
| /external/toybox/kconfig/ |
| D | conf.c | 29 } input_mode = ask_all; variable
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/ |
| D | cudnn_rnn_ops.cc | 157 int input_size, RnnInputMode* input_mode) { in ToRNNInputMode()
540 RnnInputMode input_mode; in ExtractCudnnRNNParamsInfo() local
836 RnnInputMode input_mode; in Compute() local
1037 RnnInputMode input_mode; in Compute() local
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| D | cudnn_rnn_ops_test.py | 63 input_mode="linear_input", argument
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 346 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 899 cudnnRNNInputMode_t ToCudnnRnnInputMode(dnn::RnnInputMode input_mode) { in ToCudnnRnnInputMode()
1078 cudnnRNNInputMode_t input_mode, in CudnnRnnDescriptor()
1142 cudnnRNNInputMode_t input_mode() const { return input_mode_; } in input_mode() function in perftools::gputools::cuda::CudnnRnnDescriptor
1756 int input_size, dnn::RnnInputMode input_mode, in createRnnDescriptor()
|