Home
last modified time | relevance | path

Searched defs:output_h_desc (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddnn.h2124 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward()
2146 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward()
2168 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward()
2231 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward()
2260 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward()
2289 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward()
Dstream.cc2042 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnForward()
2077 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnForward()
2113 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnForward()
2149 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnBackward()
2194 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnBackward()
2240 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2036 const MIOpenRnnStateTensorDescriptor& output_h_desc, in ExtractAndCheckRnnForward()
2146 const MIOpenRnnStateTensorDescriptor& output_h_desc, in DoRnnForwardImpl()
2261 const MIOpenRnnStateTensorDescriptor& output_h_desc, in DoRnnBackwardImpl()
2643 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward()
2686 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward()
2730 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward()
2753 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward()
2805 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward()
2858 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc1782 const CudnnRnnStateTensorDescriptor& output_h_desc, in ExtractAndCheckRnnForward()
1940 const CudnnRnnStateTensorDescriptor& output_h_desc, in DoRnnForwardImpl()
2153 const CudnnRnnStateTensorDescriptor& output_h_desc, in DoRnnBackwardImpl()
2481 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward()
2524 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward()
2568 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward()
2612 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward()
2664 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward()
2717 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward()