Searched refs:direction_mode (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 890 dnn::RnnDirectionMode direction_mode) { in ToCudnnRnnDirectionMode() argument
891 switch (direction_mode) { in ToCudnnRnnDirectionMode()
894 return static_cast<cudnnDirectionMode_t>(direction_mode); in ToCudnnRnnDirectionMode()
897 << static_cast<int>(direction_mode); in ToCudnnRnnDirectionMode()
981 cudnnDirectionMode_t direction_mode, int num_layers);
1007 cudnnDirectionMode_t direction_mode, in CudnnRnnDescriptor() argument
1021 direction_mode_(direction_mode), in CudnnRnnDescriptor()
1035 cudnnDirectionMode_t direction_mode, cudnnRNNMode_t rnn_mode, in Create() argument
1050 /*inputMode=*/input_mode, /*direction=*/direction_mode, in Create()
1081 rnn_mode, direction_mode, num_layers)); in Create()
[all …]
|
| D | cuda_dnn.h | 52 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1564 dnn::RnnDirectionMode direction_mode) { in ToMIOpenRnnDirectionMode() argument
1565 switch (direction_mode) { in ToMIOpenRnnDirectionMode()
1572 << static_cast<int>(direction_mode); in ToMIOpenRnnDirectionMode()
1684 miopenRNNDirectionMode_t direction_mode, in MIOpenRnnDescriptor() argument
1693 direction_mode_(direction_mode), in MIOpenRnnDescriptor()
1702 direction_mode /*direction*/, rnn_mode /*mode*/, in MIOpenRnnDescriptor()
1728 miopenRNNDirectionMode_t direction_mode() const { return direction_mode_; } in direction_mode() function in stream_executor::gpu::MIOpenRnnDescriptor
1912 (rnn_desc.direction_mode() == miopenRNNbidirection) ? 2 : 1; in ExtractAndCheckRnnForward()
2277 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode, in createRnnDescriptor() argument
2287 ToMIOpenRnnDirectionMode(direction_mode), ToMIOpenRnnMode(rnn_mode), in createRnnDescriptor()
|
| D | rocm_dnn.h | 50 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode,
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 383 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode, in createRnnDescriptor() argument
394 direction_mode, rnn_mode, data_type, algorithm_config, dropout, seed, in createRnnDescriptor()
|
| D | stream_executor_pimpl.h | 409 dnn::RnnInputMode input_mode, dnn::RnnDirectionMode direction_mode,
|
| D | dnn.h | 2044 dnn::RnnDirectionMode direction_mode,
|