Home
last modified time | relevance | path

Searched refs:RnnDirectionMode (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc127 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/
Dstream_executor_pimpl.h418 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode,
Ddnn.h132 enum class RnnDirectionMode { enum
2113 dnn::RnnDirectionMode direction_mode, in createRnnDescriptor()
Dstream_executor_pimpl.cc368 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc949 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()
Dcuda_dnn.h53 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1672 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()
Drocm_dnn.h87 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode,