Searched defs:input_c_desc (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1417 const CudnnRnnStateTensorDescriptor& input_c_desc, in ExtractAndCheckRnnForward()
1512 const CudnnRnnStateTensorDescriptor& input_c_desc, in DoRnnForwardImpl()
1665 const CudnnRnnStateTensorDescriptor& input_c_desc, in DoRnnBackwardImpl()
1877 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward()
1919 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward()
1960 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward()
2002 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
2054 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
2105 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1897 const MIOpenRnnStateTensorDescriptor& input_c_desc, in ExtractAndCheckRnnForward()
2007 const MIOpenRnnStateTensorDescriptor& input_c_desc, in DoRnnForwardImpl()
2122 const MIOpenRnnStateTensorDescriptor& input_c_desc, in DoRnnBackwardImpl()
2330 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward()
2373 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward()
2415 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward()
2437 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
2489 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
2540 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 4851 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnForward()
4888 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnForward()
4924 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnForward()
4961 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnBackward()
5008 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnBackward()
5054 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnBackward()
|