Searched defs:rnn_mode (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | cudnn_rnn_ops.cc | 97 string rnn_mode; in __anon20e2ac3d0302() local
145 string rnn_mode; in __anon20e2ac3d0402() local
200 string rnn_mode; in __anon20e2ac3d0502() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 155 bool has_dropout, bool is_training, RnnMode rnn_mode, in CudnnRnnParameters()
231 Status ParseRNNMode(const string& str, RnnMode* rnn_mode) { in ParseRNNMode()
480 RnnMode rnn_mode; member
1040 RnnMode rnn_mode() const { return model_types_.rnn_mode; } in rnn_mode() function in tensorflow::CudnnRNNKernelCommon
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1134 cudnnRNNMode_t ToCudnnRnnMode(dnn::RnnMode rnn_mode) { in ToCudnnRnnMode()
1238 cudnnRNNMode_t rnn_mode, cudnnDataType_t data_type, in CudnnRnnDescriptor()
1266 cudnnDirectionMode_t direction_mode, cudnnRNNMode_t rnn_mode, in Create()
1387 cudnnRNNMode_t rnn_mode() const { return rnn_mode_; } in rnn_mode() function in stream_executor::gpu::CudnnRnnDescriptor
1535 cudnnRNNDescriptor_t rnn_desc, cudnnRNNMode_t rnn_mode, in Create()
2416 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1738 miopenRNNMode_t ToMIOpenRnnMode(dnn::RnnMode rnn_mode) { in ToMIOpenRnnMode()
1820 miopenRNNMode_t rnn_mode, miopenDataType_t data_type, in MIOpenRnnDescriptor()
1864 miopenRNNMode_t rnn_mode() const { return rnn_mode_; } in rnn_mode() function in stream_executor::gpu::MIOpenRnnDescriptor
1901 auto rnn_mode = rnn_desc_->rnn_mode(); in GetRegionCountPerLayer() local
2572 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream_executor_pimpl.cc | 378 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor()
|
| D | dnn.h | 2042 dnn::RnnMode rnn_mode, dnn::DataType data_type, in createRnnDescriptor()
|