| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
| D | cudnn_rnn_ops.py | 195 input_mode=CUDNN_INPUT_LINEAR_MODE, argument
249 self._input_mode = input_mode
314 input_mode=self._input_mode,
335 input_mode=self._input_mode,
778 def check_input_mode(input_mode): argument
779 if input_mode not in (CUDNN_INPUT_LINEAR_MODE, CUDNN_INPUT_SKIP_MODE,
782 (input_mode, CUDNN_INPUT_LINEAR_MODE,
810 input_mode=CUDNN_INPUT_LINEAR_MODE, argument
846 check_input_mode(input_mode)
855 input_mode=input_mode,
[all …]
|
| /external/toybox/kconfig/ |
| D | conf.c | 29 } input_mode = ask_all; variable
59 if (!valid_stdin && input_mode == ask_silent) { in check_stdin()
85 switch (input_mode) { in conf_askvalue()
122 switch (input_mode) { in conf_askvalue()
348 switch (input_mode) { in conf_choice()
424 if (input_mode == ask_silent && rootEntry != menu) { in conf()
502 input_mode = ask_new; in main()
505 input_mode = ask_silent; in main()
509 input_mode = set_default; in main()
512 input_mode = set_default; in main()
[all …]
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/ |
| D | cudnn_rnn.py | 156 input_mode=CUDNN_INPUT_LINEAR_MODE, argument
197 cudnn_rnn_ops.check_input_mode(input_mode)
206 self._input_mode = input_mode
225 def input_mode(self): member in _CudnnRNN
462 input_mode=self._input_mode,
475 input_mode=self._input_mode,
498 input_mode=self.input_mode,
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/ |
| D | cudnn_rnn_ops.cc | 157 int input_size, RnnInputMode* input_mode) { in ToRNNInputMode() argument
160 *input_mode = RnnInputMode::kRnnLinearSkip; in ToRNNInputMode()
163 *input_mode = RnnInputMode::kRnnSkipInput; in ToRNNInputMode()
166 *input_mode = (input_size == num_units) ? RnnInputMode::kRnnSkipInput in ToRNNInputMode()
540 RnnInputMode input_mode; in ExtractCudnnRNNParamsInfo() local
542 ToRNNInputMode(rnn_input_mode(), num_units, input_size, &input_mode)); in ExtractCudnnRNNParamsInfo()
548 num_layers, num_units, input_size, input_mode, rnn_direction_mode(), in ExtractCudnnRNNParamsInfo()
836 RnnInputMode input_mode; in Compute() local
839 model_shapes.input_size, &input_mode)); in Compute()
896 model_shapes.input_size, input_mode, rnn_direction_mode(), in Compute()
[all …]
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| D | cudnn_rnn_ops_test.py | 63 input_mode="linear_input", argument
67 del input_mode
114 model.input_mode,
428 input_mode="auto_select",
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 899 cudnnRNNInputMode_t ToCudnnRnnInputMode(dnn::RnnInputMode input_mode) { in ToCudnnRnnInputMode() argument
900 switch (input_mode) { in ToCudnnRnnInputMode()
903 return static_cast<cudnnRNNInputMode_t>(input_mode); in ToCudnnRnnInputMode()
905 LOG(FATAL) << "Invalid RNN input mode: " << static_cast<int>(input_mode); in ToCudnnRnnInputMode()
1078 cudnnRNNInputMode_t input_mode, in CudnnRnnDescriptor() argument
1088 input_mode_(input_mode), in CudnnRnnDescriptor()
1109 input_mode /*inputMode*/, direction_mode /*direction*/, in CudnnRnnDescriptor()
1115 input_mode /*inputMode*/, direction_mode /*direction*/, 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() argument
[all …]
|
| D | cuda_dnn.h | 52 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode,
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 346 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode, in createRnnDescriptor() argument
355 num_layers, hidden_size, input_size, input_mode, direction_mode, rnn_mode, in createRnnDescriptor()
|
| D | stream_executor_pimpl.h | 374 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode,
|
| D | dnn.h | 1975 dnn::RnnInputMode input_mode,
|