Searched refs:RnnDirectionMode (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 127 using se::dnn::RnnDirectionMode;
260 RnnDirectionMode* rnn_dir_mode) { in ParseRNNDirectionMode()
262 *rnn_dir_mode = RnnDirectionMode::kRnnUnidirectional; in ParseRNNDirectionMode()
265 *rnn_dir_mode = RnnDirectionMode::kRnnBidirectional; in ParseRNNDirectionMode()
479 RnnDirectionMode rnn_direction_mode;
591 (model_types.rnn_direction_mode == RnnDirectionMode::kRnnBidirectional) in ExtractForwardInput()
1000 RnnDirectionMode rnn_direction_mode() const { in rnn_direction_mode()
1225 if (rnn_direction_mode() == RnnDirectionMode::kRnnBidirectional) { in Compute()
1271 if (rnn_direction_mode() == RnnDirectionMode::kRnnUnidirectional) { in Compute()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.h | 418 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode,
|
| D | dnn.h | 132 enum class RnnDirectionMode { enum
2113 dnn::RnnDirectionMode direction_mode, in createRnnDescriptor()
|
| D | stream_executor_pimpl.cc | 368 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 949 dnn::RnnDirectionMode direction_mode) { in ToCudnnRnnDirectionMode()
951 case dnn::RnnDirectionMode::kRnnUnidirectional: in ToCudnnRnnDirectionMode()
952 case dnn::RnnDirectionMode::kRnnBidirectional: in ToCudnnRnnDirectionMode()
2073 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor()
|
| D | cuda_dnn.h | 53 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1672 dnn::RnnDirectionMode direction_mode) { in ToMIOpenRnnDirectionMode()
1674 case dnn::RnnDirectionMode::kRnnUnidirectional: in ToMIOpenRnnDirectionMode()
1676 case dnn::RnnDirectionMode::kRnnBidirectional: in ToMIOpenRnnDirectionMode()
2518 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor()
|
| D | rocm_dnn.h | 87 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode,
|