Searched defs:rnn_mode (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | cudnn_rnn_ops.cc | 97 string rnn_mode; in __anonb0821bee0302() local
145 string rnn_mode; in __anonb0821bee0402() local
200 string rnn_mode; in __anonb0821bee0502() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | stream_executor_pimpl.cc | 378 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor()
|
| H A D | dnn.h | 2042 dnn::RnnMode rnn_mode, dnn::DataType data_type, in createRnnDescriptor()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/miopen/ |
| H A D | RNN_miopen.cpp | 76 miopenRNNMode_t rnn_mode; member
|