Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h2117 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2139 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2161 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2224 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
2253 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
2282 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
H A Dstream.cc2035 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnForward()
2071 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnForward()
2106 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnForward()
2142 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnBackward()
2188 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnBackward()
2233 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnBackward()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc2030 const MIOpenRnnStateTensorDescriptor& input_h_desc, in ExtractAndCheckRnnForward()
2140 const MIOpenRnnStateTensorDescriptor& input_h_desc, in DoRnnForwardImpl()
2255 const MIOpenRnnStateTensorDescriptor& input_h_desc, in DoRnnBackwardImpl()
2636 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2680 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2723 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2746 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
2799 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
2851 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc1776 const CudnnRnnStateTensorDescriptor& input_h_desc, in ExtractAndCheckRnnForward()
1934 const CudnnRnnStateTensorDescriptor& input_h_desc, in DoRnnForwardImpl()
2147 const CudnnRnnStateTensorDescriptor& input_h_desc, in DoRnnBackwardImpl()
2474 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2518 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2561 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward()
2605 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
2658 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()
2710 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward()