Searched refs:direction_mode (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 949 dnn::RnnDirectionMode direction_mode) { in ToCudnnRnnDirectionMode() argument
950 switch (direction_mode) { in ToCudnnRnnDirectionMode()
953 return static_cast<cudnnDirectionMode_t>(direction_mode); in ToCudnnRnnDirectionMode()
956 << static_cast<int>(direction_mode); in ToCudnnRnnDirectionMode()
1040 cudnnDirectionMode_t direction_mode, int num_layers);
1062 cudnnDirectionMode_t direction_mode, in CudnnRnnDescriptor() argument
1077 direction_mode_(direction_mode), in CudnnRnnDescriptor()
1091 cudnnDirectionMode_t direction_mode, cudnnRNNMode_t rnn_mode, in Create() argument
1140 /*biasMode=*/bias_mode, /*dirMode=*/direction_mode, in Create()
1154 /*direction=*/direction_mode, /*mode=*/rnn_mode, /*algo=*/rnn_algo, in Create()
[all …]
|
| D | cuda_dnn.h | 53 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode,
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1672 dnn::RnnDirectionMode direction_mode) { in ToMIOpenRnnDirectionMode() argument
1673 switch (direction_mode) { in ToMIOpenRnnDirectionMode()
1680 << static_cast<int>(direction_mode); in ToMIOpenRnnDirectionMode()
1765 miopenRNNDirectionMode_t direction_mode, in MIOpenRnnDescriptor() argument
1774 direction_mode_(direction_mode), in MIOpenRnnDescriptor()
1783 direction_mode /*direction*/, rnn_mode /*mode*/, in MIOpenRnnDescriptor()
1809 miopenRNNDirectionMode_t direction_mode() const { return direction_mode_; } in direction_mode() function in stream_executor::gpu::MIOpenRnnDescriptor
1993 (rnn_desc.direction_mode() == miopenRNNbidirection) ? 2 : 1; in ExtractAndCheckRnnForward()
2518 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor() argument
2540 ToMIOpenRnnDirectionMode(direction_mode), ToMIOpenRnnMode(rnn_mode), in createRnnDescriptor()
|
| D | rocm_dnn.h | 87 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode,
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 368 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode, in createRnnDescriptor() argument
379 direction_mode, rnn_mode, data_type, algorithm_config, dropout, seed, in createRnnDescriptor()
|
| D | stream_executor_pimpl.h | 418 dnn::RnnDirectionMode direction_mode, dnn::RnnMode rnn_mode,
|
| D | dnn.h | 2113 dnn::RnnDirectionMode direction_mode, in createRnnDescriptor() argument
|