Searched defs:direction_mode (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 949 dnn::RnnDirectionMode direction_mode) { in ToCudnnRnnDirectionMode()
1062 cudnnDirectionMode_t direction_mode, in CudnnRnnDescriptor()
1091 cudnnDirectionMode_t direction_mode, cudnnRNNMode_t rnn_mode, in Create()
1208 cudnnDirectionMode_t direction_mode() const { return direction_mode_; } in direction_mode() function in stream_executor::gpu::CudnnRnnDescriptor
1358 cudnnDirectionMode_t direction_mode, int num_layers) { in Create()
2073 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 | 2113 dnn::RnnDirectionMode direction_mode, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1672 dnn::RnnDirectionMode direction_mode) { in ToMIOpenRnnDirectionMode()
1765 miopenRNNDirectionMode_t direction_mode, in MIOpenRnnDescriptor()
1809 miopenRNNDirectionMode_t direction_mode() const { return direction_mode_; } in direction_mode() function in stream_executor::gpu::MIOpenRnnDescriptor
2518 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor()
|