Searched defs:rnn_mode (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | cudnn_rnn_ops.cc | 91 string rnn_mode; in __anon815c5c9f0302() local
134 string rnn_mode; in __anon815c5c9f0402() local
182 string rnn_mode; in __anon815c5c9f0502() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 152 bool has_dropout, bool is_training, RnnMode rnn_mode, in CudnnRnnParameters()
228 Status ParseRNNMode(const string& str, RnnMode* rnn_mode) { in ParseRNNMode()
477 RnnMode rnn_mode; member
998 RnnMode rnn_mode() const { return model_types_.rnn_mode; } in rnn_mode() function in tensorflow::CudnnRNNKernelCommon
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 960 cudnnRNNMode_t ToCudnnRnnMode(dnn::RnnMode rnn_mode) { in ToCudnnRnnMode()
1063 cudnnRNNMode_t rnn_mode, cudnnDataType_t data_type, in CudnnRnnDescriptor()
1091 cudnnDirectionMode_t direction_mode, cudnnRNNMode_t rnn_mode, in Create()
1209 cudnnRNNMode_t rnn_mode() const { return rnn_mode_; } in rnn_mode() function in stream_executor::gpu::CudnnRnnDescriptor
1357 cudnnRNNDescriptor_t rnn_desc, cudnnRNNMode_t rnn_mode, in Create()
2073 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1684 miopenRNNMode_t ToMIOpenRnnMode(dnn::RnnMode rnn_mode) { in ToMIOpenRnnMode()
1766 miopenRNNMode_t rnn_mode, miopenDataType_t data_type, in MIOpenRnnDescriptor()
1810 miopenRNNMode_t rnn_mode() const { return rnn_mode_; } in rnn_mode() function in stream_executor::gpu::MIOpenRnnDescriptor
1847 auto rnn_mode = rnn_desc_->rnn_mode(); in GetRegionCountPerLayer() local
2518 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 368 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor()
|
| D | dnn.h | 2114 dnn::RnnMode rnn_mode, dnn::DataType data_type, in createRnnDescriptor()
|