Searched defs:rnn_mode (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/ops/ |
| D | cudnn_rnn_ops.cc | 91 string rnn_mode; in __anon4fa58c430302() local
134 string rnn_mode; in __anon4fa58c430402() local
182 string rnn_mode; in __anon4fa58c430502() 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
994 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 | 943 cudnnRNNMode_t ToCudnnRnnMode(dnn::RnnMode rnn_mode) { in ToCudnnRnnMode()
1046 cudnnRNNMode_t rnn_mode, cudnnDataType_t data_type, in CudnnRnnDescriptor()
1074 cudnnDirectionMode_t direction_mode, cudnnRNNMode_t rnn_mode, in Create()
1180 cudnnRNNMode_t rnn_mode() const { return rnn_mode_; } in rnn_mode() function in stream_executor::gpu::CudnnRnnDescriptor
1317 cudnnRNNDescriptor_t rnn_desc, cudnnRNNMode_t rnn_mode, in Create()
2070 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 357 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1620 miopenRNNMode_t ToMIOpenRnnMode(dnn::RnnMode rnn_mode) { in ToMIOpenRnnMode()
1729 miopenRNNMode_t rnn_mode, miopenDataType_t data_type, in MIOpenRnnDescriptor()
1773 miopenRNNMode_t rnn_mode() const { return rnn_mode_; } in rnn_mode() function in stream_executor::gpu::MIOpenRnnDescriptor
1810 auto rnn_mode = rnn_desc_->rnn_mode(); in GetRegionCountPerLayer() local
2322 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor()
|