Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h79 const dnn::RnnStateTensorDescriptor& input_c_desc,
97 const dnn::RnnStateTensorDescriptor& input_c_desc,
115 const dnn::RnnStateTensorDescriptor& input_c_desc,
133 const dnn::RnnStateTensorDescriptor& input_c_desc,
158 const dnn::RnnStateTensorDescriptor& input_c_desc,
183 const dnn::RnnStateTensorDescriptor& input_c_desc,
646 const CudnnRnnStateTensorDescriptor& input_c_desc,
665 const CudnnRnnStateTensorDescriptor& input_c_desc,
Dcuda_dnn.cc1603 const CudnnRnnStateTensorDescriptor& input_c_desc, in ExtractAndCheckRnnForward() argument
1631 if (!(input_h_desc.num_layers() == input_c_desc.num_layers() && in ExtractAndCheckRnnForward()
1632 input_h_desc.batch_size() == input_c_desc.batch_size() && in ExtractAndCheckRnnForward()
1633 input_h_desc.data_size() <= input_c_desc.data_size())) { in ExtractAndCheckRnnForward()
1752 const CudnnRnnStateTensorDescriptor& input_c_desc, in DoRnnForwardImpl() argument
1767 input_c_desc, input_c_data, params, output_desc, *output_data, in DoRnnForwardImpl()
1811 /*cxDesc=*/input_c_desc.handle(), /*cx=*/input_c_data.opaque(), in DoRnnForwardImpl()
1827 /*hx=*/input_h_data.opaque(), /*cxDesc=*/input_c_desc.handle(), in DoRnnForwardImpl()
1842 /*cxDesc=*/input_c_desc.handle(), /*cx=*/input_c_data.opaque(), in DoRnnForwardImpl()
1860 /*hx=*/input_h_data.opaque(), /*cxDesc=*/input_c_desc.handle(), in DoRnnForwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h106 const dnn::RnnStateTensorDescriptor& input_c_desc,
124 const dnn::RnnStateTensorDescriptor& input_c_desc,
142 const dnn::RnnStateTensorDescriptor& input_c_desc,
160 const dnn::RnnStateTensorDescriptor& input_c_desc,
185 const dnn::RnnStateTensorDescriptor& input_c_desc,
210 const dnn::RnnStateTensorDescriptor& input_c_desc,
743 const MIOpenRnnStateTensorDescriptor& input_c_desc,
760 const MIOpenRnnStateTensorDescriptor& input_c_desc,
Drocm_dnn.cc1978 const MIOpenRnnStateTensorDescriptor& input_c_desc, in ExtractAndCheckRnnForward() argument
2003 if (!(input_h_desc.num_layers() == input_c_desc.num_layers() && in ExtractAndCheckRnnForward()
2004 input_h_desc.batch_size() == input_c_desc.batch_size() && in ExtractAndCheckRnnForward()
2005 input_h_desc.data_size() == input_c_desc.data_size())) { in ExtractAndCheckRnnForward()
2088 const MIOpenRnnStateTensorDescriptor& input_c_desc, in DoRnnForwardImpl() argument
2102 input_c_desc, input_c_data, params, output_desc, *output_data, in DoRnnForwardImpl()
2160 input_h_data.opaque() /*hx*/, input_c_desc.handle() /*cxDesc*/, in DoRnnForwardImpl()
2178 input_h_data.opaque() /*hx*/, input_c_desc.handle() /*cxDesc*/, in DoRnnForwardImpl()
2203 const MIOpenRnnStateTensorDescriptor& input_c_desc, in DoRnnBackwardImpl() argument
2224 input_c_desc, input_c_data, params, output_desc, output_data, in DoRnnBackwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2190 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward() argument
2211 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward() argument
2232 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnForward() argument
2294 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward() argument
2322 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward() argument
2350 const dnn::RnnStateTensorDescriptor& input_c_desc, in DoRnnBackward() argument
Dstream.h1784 const dnn::RnnStateTensorDescriptor &input_c_desc,
1803 const dnn::RnnStateTensorDescriptor &input_c_desc,
1821 const dnn::RnnStateTensorDescriptor &input_c_desc,
1842 const dnn::RnnStateTensorDescriptor &input_c_desc,
1867 const dnn::RnnStateTensorDescriptor &input_c_desc,
1892 const dnn::RnnStateTensorDescriptor &input_c_desc,
Dstream.cc4544 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnForward() argument
4560 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward()
4578 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnForward() argument
4593 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward()
4611 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnForward() argument
4627 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward()
4645 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnBackward() argument
4668 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnBackward()
4690 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnBackward() argument
4712 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnBackward()
[all …]