Home
last modified time | relevance | path

Searched defs:input_c_desc (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1417 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/
Drocm_dnn.cc1897 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/
Dstream.cc4851 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()