Searched refs:DoRnnForward (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 73 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc, 91 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc, 109 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
|
D | cuda_dnn.cc | 1871 bool CudnnSupport::DoRnnForward( in DoRnnForward() function in stream_executor::gpu::CudnnSupport 1913 bool CudnnSupport::DoRnnForward( in DoRnnForward() function in stream_executor::gpu::CudnnSupport 1954 bool CudnnSupport::DoRnnForward( in DoRnnForward() function in stream_executor::gpu::CudnnSupport
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 64 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc, 82 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc, 100 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
|
D | rocm_dnn.cc | 2324 bool MIOpenSupport::DoRnnForward( in DoRnnForward() function in stream_executor::gpu::MIOpenSupport 2367 bool MIOpenSupport::DoRnnForward( in DoRnnForward() function in stream_executor::gpu::MIOpenSupport 2409 bool MIOpenSupport::DoRnnForward( in DoRnnForward() function in stream_executor::gpu::MIOpenSupport
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 2116 virtual bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc, 2137 virtual bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc, 2158 virtual bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
|
D | stream.cc | 4866 auto status = dnn->DoRnnForward( in ThenRnnForward() 4902 auto status = dnn->DoRnnForward( in ThenRnnForward() 4939 auto status = dnn->DoRnnForward( in ThenRnnForward()
|