Searched defs:direction_mode (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 1123 dnn::RnnDirectionMode direction_mode) { in ToCudnnRnnDirectionMode()
1237 cudnnDirectionMode_t direction_mode, in CudnnRnnDescriptor()
1266 cudnnDirectionMode_t direction_mode, cudnnRNNMode_t rnn_mode, in Create()
1386 cudnnDirectionMode_t direction_mode() const { return direction_mode_; } in direction_mode() function in stream_executor::gpu::CudnnRnnDescriptor
1536 cudnnDirectionMode_t direction_mode, int num_layers) { 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 | 1726 dnn::RnnDirectionMode direction_mode) { in ToMIOpenRnnDirectionMode()
1819 miopenRNNDirectionMode_t direction_mode, in MIOpenRnnDescriptor()
1863 miopenRNNDirectionMode_t direction_mode() const { return direction_mode_; } in direction_mode() function in stream_executor::gpu::MIOpenRnnDescriptor
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 | 2041 dnn::RnnDirectionMode direction_mode, in createRnnDescriptor()
|