Home
last modified time | relevance | path

Searched refs:DoRnnForward (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h74 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
92 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
110 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
Dcuda_dnn.cc2126 bool CudnnSupport::DoRnnForward( in DoRnnForward() function in stream_executor::gpu::CudnnSupport
2168 bool CudnnSupport::DoRnnForward( in DoRnnForward() function in stream_executor::gpu::CudnnSupport
2209 bool CudnnSupport::DoRnnForward( in DoRnnForward() function in stream_executor::gpu::CudnnSupport
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h101 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
119 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
137 bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc,
Drocm_dnn.cc2577 bool MIOpenSupport::DoRnnForward( in DoRnnForward() function in stream_executor::gpu::MIOpenSupport
2620 bool MIOpenSupport::DoRnnForward( in DoRnnForward() function in stream_executor::gpu::MIOpenSupport
2662 bool MIOpenSupport::DoRnnForward( in DoRnnForward() function in stream_executor::gpu::MIOpenSupport
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2185 virtual bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward() function
2206 virtual bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward() function
2227 virtual bool DoRnnForward(Stream* stream, const dnn::RnnDescriptor& rnn_desc, in DoRnnForward() function
Dstream.cc4558 auto status = dnn->DoRnnForward( in ThenRnnForward()
4591 auto status = dnn->DoRnnForward( in ThenRnnForward()
4625 auto status = dnn->DoRnnForward( in ThenRnnForward()