Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h2119 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward()
2141 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward()
2163 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward()
2226 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
2255 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
2284 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
H A Dstream.cc2037 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnForward()
2073 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnForward()
2108 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnForward()
2144 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnBackward()
2190 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnBackward()
2235 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnBackward()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc2032 const MIOpenRnnStateTensorDescriptor& input_c_desc, in ExtractAndCheckRnnForward()
2142 const MIOpenRnnStateTensorDescriptor& input_c_desc, in DoRnnForwardImpl()
2257 const MIOpenRnnStateTensorDescriptor& input_c_desc, in DoRnnBackwardImpl()
2638 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward()
2682 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward()
2725 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward()
2748 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
2801 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
2853 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc1778 const CudnnRnnStateTensorDescriptor& input_c_desc, in ExtractAndCheckRnnForward()
1936 const CudnnRnnStateTensorDescriptor& input_c_desc, in DoRnnForwardImpl()
2149 const CudnnRnnStateTensorDescriptor& input_c_desc, in DoRnnBackwardImpl()
2476 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward()
2520 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward()
2563 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward()
2607 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
2660 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()
2712 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward()