Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Ddnn.h2126 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2148 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2170 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2233 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
2262 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
2291 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
Dstream.cc2044 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward()
2079 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward()
2115 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward()
2151 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnBackward()
2196 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnBackward()
2242 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2038 const MIOpenRnnStateTensorDescriptor& output_c_desc, in ExtractAndCheckRnnForward()
2148 const MIOpenRnnStateTensorDescriptor& output_c_desc, in DoRnnForwardImpl()
2263 const MIOpenRnnStateTensorDescriptor& output_c_desc, in DoRnnBackwardImpl()
2645 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2688 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2732 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2755 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
2807 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
2860 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_dnn.cc1784 const CudnnRnnStateTensorDescriptor& output_c_desc, in ExtractAndCheckRnnForward()
1942 const CudnnRnnStateTensorDescriptor& output_c_desc, in DoRnnForwardImpl()
2155 const CudnnRnnStateTensorDescriptor& output_c_desc, in DoRnnBackwardImpl()
2483 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2526 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2570 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2614 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
2666 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
2719 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()