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,
|