Searched defs:input_h_desc (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1415 const CudnnRnnStateTensorDescriptor& input_h_desc, in ExtractAndCheckRnnForward()
1510 const CudnnRnnStateTensorDescriptor& input_h_desc, in DoRnnForwardImpl()
1663 const CudnnRnnStateTensorDescriptor& input_h_desc, in DoRnnBackwardImpl()
1875 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
1917 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
1958 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2000 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
2052 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
2103 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1895 const MIOpenRnnStateTensorDescriptor& input_h_desc, in ExtractAndCheckRnnForward()
2005 const MIOpenRnnStateTensorDescriptor& input_h_desc, in DoRnnForwardImpl()
2120 const MIOpenRnnStateTensorDescriptor& input_h_desc, in DoRnnBackwardImpl()
2328 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2371 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2413 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2435 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
2487 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
2538 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 4849 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnForward()
4886 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnForward()
4922 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnForward()
4959 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnBackward()
5006 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnBackward()
5052 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnBackward()
|