Home
last modified time | relevance | path

Searched refs:output_h_desc (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h84 const dnn::RnnStateTensorDescriptor& output_h_desc,
102 const dnn::RnnStateTensorDescriptor& output_h_desc,
120 const dnn::RnnStateTensorDescriptor& output_h_desc,
138 const dnn::RnnStateTensorDescriptor& output_h_desc,
163 const dnn::RnnStateTensorDescriptor& output_h_desc,
188 const dnn::RnnStateTensorDescriptor& output_h_desc,
650 const CudnnRnnStateTensorDescriptor& output_h_desc,
669 const CudnnRnnStateTensorDescriptor& output_h_desc,
Dcuda_dnn.cc1607 const CudnnRnnStateTensorDescriptor& output_h_desc, in ExtractAndCheckRnnForward() argument
1642 if (!(input_h_desc.num_layers() == output_h_desc.num_layers() && in ExtractAndCheckRnnForward()
1643 input_h_desc.batch_size() == output_h_desc.batch_size() && in ExtractAndCheckRnnForward()
1644 input_h_desc.data_size() == output_h_desc.data_size())) { in ExtractAndCheckRnnForward()
1756 const CudnnRnnStateTensorDescriptor& output_h_desc, in DoRnnForwardImpl() argument
1768 output_h_desc, *output_h_data, output_c_desc, *output_c_data)); in DoRnnForwardImpl()
1815 /*hyDesc=*/output_h_desc.handle(), /*hy=*/output_h_data->opaque(), in DoRnnForwardImpl()
1830 /*y=*/output_data->opaque(), /*hyDesc=*/output_h_desc.handle(), in DoRnnForwardImpl()
1846 /*hyDesc=*/output_h_desc.handle(), /*hy=*/output_h_data->opaque(), in DoRnnForwardImpl()
1863 /*y=*/output_data->opaque(), /*hyDesc=*/output_h_desc.handle(), in DoRnnForwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h111 const dnn::RnnStateTensorDescriptor& output_h_desc,
129 const dnn::RnnStateTensorDescriptor& output_h_desc,
147 const dnn::RnnStateTensorDescriptor& output_h_desc,
165 const dnn::RnnStateTensorDescriptor& output_h_desc,
190 const dnn::RnnStateTensorDescriptor& output_h_desc,
215 const dnn::RnnStateTensorDescriptor& output_h_desc,
748 const MIOpenRnnStateTensorDescriptor& output_h_desc,
765 const MIOpenRnnStateTensorDescriptor& output_h_desc,
Drocm_dnn.cc1982 const MIOpenRnnStateTensorDescriptor& output_h_desc, in ExtractAndCheckRnnForward() argument
2016 if (!(input_h_desc.num_layers() == output_h_desc.num_layers() && in ExtractAndCheckRnnForward()
2017 input_h_desc.batch_size() == output_h_desc.batch_size() && in ExtractAndCheckRnnForward()
2018 input_h_desc.data_size() == output_h_desc.data_size())) { in ExtractAndCheckRnnForward()
2092 const MIOpenRnnStateTensorDescriptor& output_h_desc, in DoRnnForwardImpl() argument
2103 output_h_desc, *output_h_data, output_c_desc, *output_c_data, in DoRnnForwardImpl()
2163 output_data->opaque() /*y*/, output_h_desc.handle() /*hyDesc*/, in DoRnnForwardImpl()
2181 output_data->opaque() /*y*/, output_h_desc.handle() /*hyDesc*/, in DoRnnForwardImpl()
2207 const MIOpenRnnStateTensorDescriptor& output_h_desc, in DoRnnBackwardImpl() argument
2225 output_h_desc, output_h_data, output_c_desc, output_c_data, &model_dims); in DoRnnBackwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2195 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward() argument
2216 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward() argument
2237 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnForward() argument
2299 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward() argument
2327 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward() argument
2355 const dnn::RnnStateTensorDescriptor& output_h_desc, in DoRnnBackward() argument
Dstream.h1789 const dnn::RnnStateTensorDescriptor &output_h_desc,
1808 const dnn::RnnStateTensorDescriptor &output_h_desc,
1826 const dnn::RnnStateTensorDescriptor &output_h_desc,
1847 const dnn::RnnStateTensorDescriptor &output_h_desc,
1872 const dnn::RnnStateTensorDescriptor &output_h_desc,
1897 const dnn::RnnStateTensorDescriptor &output_h_desc,
Dstream.cc4549 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnForward() argument
4561 output_h_desc, output_h_data, output_c_desc, output_c_data, is_training, in ThenRnnForward()
4582 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnForward() argument
4594 output_h_desc, output_h_data, output_c_desc, output_c_data, is_training, in ThenRnnForward()
4616 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnForward() argument
4628 output_h_desc, output_h_data, output_c_desc, output_c_data, is_training, in ThenRnnForward()
4650 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnBackward() argument
4669 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward()
4694 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnBackward() argument
4713 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward()
[all …]