Home
last modified time | relevance | path

Searched defs:input_h_desc (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2188 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2209 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2230 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2292 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
2320 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
2348 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
Dstream.cc4542 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnForward()
4576 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnForward()
4609 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnForward()
4643 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnBackward()
4688 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnBackward()
4732 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnBackward()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1601 const CudnnRnnStateTensorDescriptor& input_h_desc, in ExtractAndCheckRnnForward()
1750 const CudnnRnnStateTensorDescriptor& input_h_desc, in DoRnnForwardImpl()
1890 const CudnnRnnStateTensorDescriptor& input_h_desc, in DoRnnBackwardImpl()
2130 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2172 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2213 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2255 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
2307 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
2358 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1976 const MIOpenRnnStateTensorDescriptor& input_h_desc, in ExtractAndCheckRnnForward()
2086 const MIOpenRnnStateTensorDescriptor& input_h_desc, in DoRnnForwardImpl()
2201 const MIOpenRnnStateTensorDescriptor& input_h_desc, in DoRnnBackwardImpl()
2581 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2624 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2666 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2688 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
2740 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
2791 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()