Searched refs:RnnInputMode (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 128 using se::dnn::RnnInputMode;
272 int input_size, RnnInputMode* input_mode) { in ToRNNInputMode()
275 *input_mode = RnnInputMode::kRnnLinearSkip; in ToRNNInputMode()
278 *input_mode = RnnInputMode::kRnnSkipInput; in ToRNNInputMode()
281 *input_mode = (input_size == num_units) ? RnnInputMode::kRnnSkipInput in ToRNNInputMode()
282 : RnnInputMode::kRnnLinearSkip; in ToRNNInputMode()
1033 RnnInputMode input_mode; in ExtractCudnnRNNParamsInfo()
1056 const RnnInputMode& input_mode, in CreateRnnDescriptor()
1083 const RnnInputMode& input_mode, in GetCachedRnnDescriptor()
1478 RnnInputMode input_mode; in ComputeAndReturnAlgorithm()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.h | 417 int batch_size, dnn::RnnInputMode input_mode,
|
| D | dnn.h | 124 enum class RnnInputMode { enum
2112 dnn::RnnInputMode input_mode, in createRnnDescriptor()
|
| D | stream_executor_pimpl.cc | 367 int batch_size, dnn::RnnInputMode input_mode, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 938 cudnnRNNInputMode_t ToCudnnRnnInputMode(dnn::RnnInputMode input_mode) { in ToCudnnRnnInputMode()
940 case dnn::RnnInputMode::kRnnLinearSkip: in ToCudnnRnnInputMode()
941 case dnn::RnnInputMode::kRnnSkipInput: in ToCudnnRnnInputMode()
2072 int batch_size, dnn::RnnInputMode input_mode, in createRnnDescriptor()
|
| D | cuda_dnn.h | 52 int batch_size, dnn::RnnInputMode input_mode,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1660 miopenRNNInputMode_t ToMIOpenRnnInputMode(dnn::RnnInputMode input_mode) { in ToMIOpenRnnInputMode()
1662 case dnn::RnnInputMode::kRnnLinearSkip: in ToMIOpenRnnInputMode()
1664 case dnn::RnnInputMode::kRnnSkipInput: in ToMIOpenRnnInputMode()
2517 int batch_size, dnn::RnnInputMode input_mode, in createRnnDescriptor()
|
| D | rocm_dnn.h | 86 int batch_size, dnn::RnnInputMode input_mode,
|