Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2197 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2218 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2239 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2301 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
2329 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
2357 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
Dstream.cc4551 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward()
4584 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward()
4618 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward()
4652 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnBackward()
4696 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnBackward()
4741 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnBackward()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1609 const CudnnRnnStateTensorDescriptor& output_c_desc, in ExtractAndCheckRnnForward()
1758 const CudnnRnnStateTensorDescriptor& output_c_desc, in DoRnnForwardImpl()
1898 const CudnnRnnStateTensorDescriptor& output_c_desc, in DoRnnBackwardImpl()
2139 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2180 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2222 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2264 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
2315 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
2367 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1984 const MIOpenRnnStateTensorDescriptor& output_c_desc, in ExtractAndCheckRnnForward()
2094 const MIOpenRnnStateTensorDescriptor& output_c_desc, in DoRnnForwardImpl()
2209 const MIOpenRnnStateTensorDescriptor& output_c_desc, in DoRnnBackwardImpl()
2590 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2632 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2675 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward()
2697 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
2748 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()
2800 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward()