| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.h | 94 const dnn::RnnStateTensorDescriptor& output_c_desc,
113 const dnn::RnnStateTensorDescriptor& output_c_desc,
132 const dnn::RnnStateTensorDescriptor& output_c_desc,
151 const dnn::RnnStateTensorDescriptor& output_c_desc,
177 const dnn::RnnStateTensorDescriptor& output_c_desc,
203 const dnn::RnnStateTensorDescriptor& output_c_desc,
580 const CudnnRnnStateTensorDescriptor& output_c_desc,
600 const CudnnRnnStateTensorDescriptor& output_c_desc,
|
| H A D | cuda_dnn.cc | 1784 const CudnnRnnStateTensorDescriptor& output_c_desc, in ExtractAndCheckRnnForward() argument
1823 if (!(input_h_desc.num_layers() == output_c_desc.num_layers() && in ExtractAndCheckRnnForward()
1824 input_h_desc.batch_size() == output_c_desc.batch_size() && in ExtractAndCheckRnnForward()
1825 input_h_desc.data_size() <= output_c_desc.data_size())) { in ExtractAndCheckRnnForward()
1942 const CudnnRnnStateTensorDescriptor& output_c_desc, in DoRnnForwardImpl() argument
1952 output_h_desc, *output_h_data, output_c_desc, *output_c_data)); in DoRnnForwardImpl()
2073 /*cyDesc=*/output_c_desc.handle(), /*cy=*/output_c_data->opaque(), in DoRnnForwardImpl()
2088 /*hy=*/output_h_data->opaque(), /*cyDesc=*/output_c_desc.handle(), in DoRnnForwardImpl()
2103 /*cyDesc=*/output_c_desc.handle(), /*cy=*/output_c_data->opaque(), in DoRnnForwardImpl()
2120 /*hy=*/output_h_data->opaque(), /*cyDesc=*/output_c_desc.handle(), in DoRnnForwardImpl()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.h | 114 const dnn::RnnStateTensorDescriptor& output_c_desc,
133 const dnn::RnnStateTensorDescriptor& output_c_desc,
152 const dnn::RnnStateTensorDescriptor& output_c_desc,
171 const dnn::RnnStateTensorDescriptor& output_c_desc,
197 const dnn::RnnStateTensorDescriptor& output_c_desc,
223 const dnn::RnnStateTensorDescriptor& output_c_desc,
678 const MIOpenRnnStateTensorDescriptor& output_c_desc,
695 const MIOpenRnnStateTensorDescriptor& output_c_desc,
|
| H A D | rocm_dnn.cc | 2038 const MIOpenRnnStateTensorDescriptor& output_c_desc, in ExtractAndCheckRnnForward() argument
2076 if (!(input_h_desc.num_layers() == output_c_desc.num_layers() && in ExtractAndCheckRnnForward()
2077 input_h_desc.batch_size() == output_c_desc.batch_size() && in ExtractAndCheckRnnForward()
2078 input_h_desc.data_size() == output_c_desc.data_size())) { in ExtractAndCheckRnnForward()
2148 const MIOpenRnnStateTensorDescriptor& output_c_desc, in DoRnnForwardImpl() argument
2157 output_h_desc, *output_h_data, output_c_desc, *output_c_data, in DoRnnForwardImpl()
2218 output_h_data->opaque() /*hy*/, output_c_desc.handle() /*cyDesc*/, in DoRnnForwardImpl()
2236 output_h_data->opaque() /*hy*/, output_c_desc.handle() /*cyDesc*/, in DoRnnForwardImpl()
2263 const MIOpenRnnStateTensorDescriptor& output_c_desc, in DoRnnBackwardImpl() argument
2279 output_h_desc, output_h_data, output_c_desc, output_c_data, &model_dims); in DoRnnBackwardImpl()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | stream.cc | 2044 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward() argument
2054 output_data, output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward()
2079 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward() argument
2089 output_data, output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward()
2115 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward() argument
2125 output_data, output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward()
2151 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnBackward() argument
2168 output_data, output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward()
2196 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnBackward() argument
2213 output_data, output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward()
[all …]
|
| H A D | dnn.h | 2126 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward() argument
2148 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward() argument
2170 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward() argument
2233 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward() argument
2262 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward() argument
2291 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward() argument
|
| H A D | stream.h | 1239 const dnn::RnnStateTensorDescriptor &output_c_desc,
1259 const dnn::RnnStateTensorDescriptor &output_c_desc,
1278 const dnn::RnnStateTensorDescriptor &output_c_desc,
1300 const dnn::RnnStateTensorDescriptor &output_c_desc,
1326 const dnn::RnnStateTensorDescriptor &output_c_desc,
1352 const dnn::RnnStateTensorDescriptor &output_c_desc,
|