Searched defs:output_c_desc (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1423 const CudnnRnnStateTensorDescriptor& output_c_desc, in ExtractAndCheckRnnForward()
1518 const CudnnRnnStateTensorDescriptor& output_c_desc, in DoRnnForwardImpl()
1671 const CudnnRnnStateTensorDescriptor& output_c_desc, in DoRnnBackwardImpl()
1884 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
1925 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
1967 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2009 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
2060 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
2112 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1903 const MIOpenRnnStateTensorDescriptor& output_c_desc, in ExtractAndCheckRnnForward()
2013 const MIOpenRnnStateTensorDescriptor& output_c_desc, in DoRnnForwardImpl()
2128 const MIOpenRnnStateTensorDescriptor& output_c_desc, in DoRnnBackwardImpl()
2337 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2379 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2422 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2444 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
2495 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
2547 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 4858 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward()
4894 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward()
4931 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward()
4968 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnBackward()
5014 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnBackward()
5061 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnBackward()
|