Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h85 const dnn::RnnStateTensorDescriptor& output_c_desc,
103 const dnn::RnnStateTensorDescriptor& output_c_desc,
121 const dnn::RnnStateTensorDescriptor& output_c_desc,
139 const dnn::RnnStateTensorDescriptor& output_c_desc,
164 const dnn::RnnStateTensorDescriptor& output_c_desc,
189 const dnn::RnnStateTensorDescriptor& output_c_desc,
618 const CudnnRnnStateTensorDescriptor& output_c_desc,
637 const CudnnRnnStateTensorDescriptor& output_c_desc,
Dcuda_dnn.cc1423 const CudnnRnnStateTensorDescriptor& output_c_desc, in ExtractAndCheckRnnForward() argument
1459 if (!(input_h_desc.num_layers() == output_c_desc.num_layers() && in ExtractAndCheckRnnForward()
1460 input_h_desc.batch_size() == output_c_desc.batch_size() && in ExtractAndCheckRnnForward()
1461 input_h_desc.data_size() == output_c_desc.data_size())) { in ExtractAndCheckRnnForward()
1518 const CudnnRnnStateTensorDescriptor& output_c_desc, in DoRnnForwardImpl() argument
1528 output_h_desc, *output_h_data, output_c_desc, *output_c_data)); in DoRnnForwardImpl()
1578 /*cyDesc=*/output_c_desc.handle(), /*cy=*/output_c_data->opaque(), in DoRnnForwardImpl()
1598 /*hy=*/output_h_data->opaque(), /*cyDesc=*/output_c_desc.handle(), in DoRnnForwardImpl()
1615 /*cyDesc=*/output_c_desc.handle(), /*cy=*/output_c_data->opaque(), in DoRnnForwardImpl()
1637 /*hy=*/output_h_data->opaque(), /*cyDesc=*/output_c_desc.handle(), in DoRnnForwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h76 const dnn::RnnStateTensorDescriptor& output_c_desc,
94 const dnn::RnnStateTensorDescriptor& output_c_desc,
112 const dnn::RnnStateTensorDescriptor& output_c_desc,
130 const dnn::RnnStateTensorDescriptor& output_c_desc,
155 const dnn::RnnStateTensorDescriptor& output_c_desc,
180 const dnn::RnnStateTensorDescriptor& output_c_desc,
675 const MIOpenRnnStateTensorDescriptor& output_c_desc,
692 const MIOpenRnnStateTensorDescriptor& output_c_desc,
Drocm_dnn.cc1903 const MIOpenRnnStateTensorDescriptor& output_c_desc, in ExtractAndCheckRnnForward() argument
1941 if (!(input_h_desc.num_layers() == output_c_desc.num_layers() && in ExtractAndCheckRnnForward()
1942 input_h_desc.batch_size() == output_c_desc.batch_size() && in ExtractAndCheckRnnForward()
1943 input_h_desc.data_size() == output_c_desc.data_size())) { in ExtractAndCheckRnnForward()
2013 const MIOpenRnnStateTensorDescriptor& output_c_desc, in DoRnnForwardImpl() argument
2022 output_h_desc, *output_h_data, output_c_desc, *output_c_data, in DoRnnForwardImpl()
2083 output_h_data->opaque() /*hy*/, output_c_desc.handle() /*cyDesc*/, in DoRnnForwardImpl()
2101 output_h_data->opaque() /*hy*/, output_c_desc.handle() /*cyDesc*/, in DoRnnForwardImpl()
2128 const MIOpenRnnStateTensorDescriptor& output_c_desc, in DoRnnBackwardImpl() argument
2144 output_h_desc, output_h_data, output_c_desc, output_c_data, &model_dims); in DoRnnBackwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2128 const dnn::RnnStateTensorDescriptor& output_c_desc,
2149 const dnn::RnnStateTensorDescriptor& output_c_desc,
2170 const dnn::RnnStateTensorDescriptor& output_c_desc,
2232 const dnn::RnnStateTensorDescriptor& output_c_desc,
2260 const dnn::RnnStateTensorDescriptor& output_c_desc,
2288 const dnn::RnnStateTensorDescriptor& output_c_desc,
Dstream.h1749 const dnn::RnnStateTensorDescriptor &output_c_desc,
1768 const dnn::RnnStateTensorDescriptor &output_c_desc,
1786 const dnn::RnnStateTensorDescriptor &output_c_desc,
1807 const dnn::RnnStateTensorDescriptor &output_c_desc,
1832 const dnn::RnnStateTensorDescriptor &output_c_desc,
1857 const dnn::RnnStateTensorDescriptor &output_c_desc,
Dstream.cc4858 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward() argument
4869 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward()
4894 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward() argument
4905 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward()
4931 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward() argument
4942 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward()
4968 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnBackward() argument
4986 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward()
5014 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnBackward() argument
5032 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward()
[all …]