Searched defs:output_c_desc (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 2197 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()
|
| D | stream.cc | 4551 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/ |
| D | cuda_dnn.cc | 1609 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/ |
| D | rocm_dnn.cc | 1984 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()
|