Searched defs:direction_mode (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 890 dnn::RnnDirectionMode direction_mode) { in ToCudnnRnnDirectionMode()
1007 cudnnDirectionMode_t direction_mode, in CudnnRnnDescriptor()
1035 cudnnDirectionMode_t direction_mode, cudnnRNNMode_t rnn_mode, in Create()
1113 cudnnDirectionMode_t direction_mode() const { return direction_mode_; } in direction_mode() function in stream_executor::gpu::CudnnRnnDescriptor
1159 cudnnDirectionMode_t direction_mode, int num_layers) { in Create()
1818 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 383 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode, in createRnnDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1564 dnn::RnnDirectionMode direction_mode) { in ToMIOpenRnnDirectionMode()
1684 miopenRNNDirectionMode_t direction_mode, in MIOpenRnnDescriptor()
1728 miopenRNNDirectionMode_t direction_mode() const { return direction_mode_; } in direction_mode() function in stream_executor::gpu::MIOpenRnnDescriptor
2277 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode, in createRnnDescriptor()
|