Searched defs:output_h_desc (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1421 const CudnnRnnStateTensorDescriptor& output_h_desc, in ExtractAndCheckRnnForward()
1516 const CudnnRnnStateTensorDescriptor& output_h_desc, in DoRnnForwardImpl()
1669 const CudnnRnnStateTensorDescriptor& output_h_desc, in DoRnnBackwardImpl()
1882 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward()
1923 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward()
1965 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward()
2007 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward()
2058 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward()
2110 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1901 const MIOpenRnnStateTensorDescriptor& output_h_desc, in ExtractAndCheckRnnForward()
2011 const MIOpenRnnStateTensorDescriptor& output_h_desc, in DoRnnForwardImpl()
2126 const MIOpenRnnStateTensorDescriptor& output_h_desc, in DoRnnBackwardImpl()
2335 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward()
2377 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward()
2420 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward()
2442 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward()
2493 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward()
2545 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 4856 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnForward()
4892 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnForward()
4929 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnForward()
4966 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnBackward()
5012 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnBackward()
5059 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnBackward()
|