Searched defs:input_c_desc (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 2190 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward() 2211 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward() 2232 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward() 2294 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward() 2322 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward() 2350 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
|
D | stream.cc | 4544 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnForward() 4578 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnForward() 4611 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnForward() 4645 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnBackward() 4690 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnBackward() 4734 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnBackward()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.cc | 1603 const CudnnRnnStateTensorDescriptor& input_c_desc, in ExtractAndCheckRnnForward() 1752 const CudnnRnnStateTensorDescriptor& input_c_desc, in DoRnnForwardImpl() 1892 const CudnnRnnStateTensorDescriptor& input_c_desc, in DoRnnBackwardImpl() 2132 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward() 2174 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward() 2215 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward() 2257 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward() 2309 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward() 2360 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.cc | 1978 const MIOpenRnnStateTensorDescriptor& input_c_desc, in ExtractAndCheckRnnForward() 2088 const MIOpenRnnStateTensorDescriptor& input_c_desc, in DoRnnForwardImpl() 2203 const MIOpenRnnStateTensorDescriptor& input_c_desc, in DoRnnBackwardImpl() 2583 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward() 2626 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward() 2668 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward() 2690 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward() 2742 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward() 2793 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
|