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.h78 const dnn::RnnStateTensorDescriptor& input_c_desc,
96 const dnn::RnnStateTensorDescriptor& input_c_desc,
114 const dnn::RnnStateTensorDescriptor& input_c_desc,
132 const dnn::RnnStateTensorDescriptor& input_c_desc,
157 const dnn::RnnStateTensorDescriptor& input_c_desc,
182 const dnn::RnnStateTensorDescriptor& input_c_desc,
612 const CudnnRnnStateTensorDescriptor& input_c_desc,
631 const CudnnRnnStateTensorDescriptor& input_c_desc,
Dcuda_dnn.cc1417 const CudnnRnnStateTensorDescriptor& input_c_desc, in ExtractAndCheckRnnForward() argument
1442 if (!(input_h_desc.num_layers() == input_c_desc.num_layers() && in ExtractAndCheckRnnForward()
1443 input_h_desc.batch_size() == input_c_desc.batch_size() && in ExtractAndCheckRnnForward()
1444 input_h_desc.data_size() == input_c_desc.data_size())) { in ExtractAndCheckRnnForward()
1512 const CudnnRnnStateTensorDescriptor& input_c_desc, in DoRnnForwardImpl() argument
1527 input_c_desc, input_c_data, params, output_desc, *output_data, in DoRnnForwardImpl()
1573 /*cxDesc=*/input_c_desc.handle(), /*cx=*/input_c_data.opaque(), in DoRnnForwardImpl()
1594 /*hx=*/input_h_data.opaque(), /*cxDesc=*/input_c_desc.handle(), in DoRnnForwardImpl()
1610 /*cxDesc=*/input_c_desc.handle(), /*cx=*/input_c_data.opaque(), in DoRnnForwardImpl()
1633 /*hx=*/input_h_data.opaque(), /*cxDesc=*/input_c_desc.handle(), in DoRnnForwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h69 const dnn::RnnStateTensorDescriptor& input_c_desc,
87 const dnn::RnnStateTensorDescriptor& input_c_desc,
105 const dnn::RnnStateTensorDescriptor& input_c_desc,
123 const dnn::RnnStateTensorDescriptor& input_c_desc,
148 const dnn::RnnStateTensorDescriptor& input_c_desc,
173 const dnn::RnnStateTensorDescriptor& input_c_desc,
668 const MIOpenRnnStateTensorDescriptor& input_c_desc,
685 const MIOpenRnnStateTensorDescriptor& input_c_desc,
Drocm_dnn.cc1897 const MIOpenRnnStateTensorDescriptor& input_c_desc, in ExtractAndCheckRnnForward() argument
1922 if (!(input_h_desc.num_layers() == input_c_desc.num_layers() && in ExtractAndCheckRnnForward()
1923 input_h_desc.batch_size() == input_c_desc.batch_size() && in ExtractAndCheckRnnForward()
1924 input_h_desc.data_size() == input_c_desc.data_size())) { in ExtractAndCheckRnnForward()
2007 const MIOpenRnnStateTensorDescriptor& input_c_desc, in DoRnnForwardImpl() argument
2021 input_c_desc, input_c_data, params, output_desc, *output_data, in DoRnnForwardImpl()
2079 input_h_data.opaque() /*hx*/, input_c_desc.handle() /*cxDesc*/, in DoRnnForwardImpl()
2097 input_h_data.opaque() /*hx*/, input_c_desc.handle() /*cxDesc*/, in DoRnnForwardImpl()
2122 const MIOpenRnnStateTensorDescriptor& input_c_desc, in DoRnnBackwardImpl() argument
2143 input_c_desc, input_c_data, params, output_desc, output_data, in DoRnnBackwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2121 const dnn::RnnStateTensorDescriptor& input_c_desc,
2142 const dnn::RnnStateTensorDescriptor& input_c_desc,
2163 const dnn::RnnStateTensorDescriptor& input_c_desc,
2225 const dnn::RnnStateTensorDescriptor& input_c_desc,
2253 const dnn::RnnStateTensorDescriptor& input_c_desc,
2281 const dnn::RnnStateTensorDescriptor& input_c_desc,
Dstream.h1742 const dnn::RnnStateTensorDescriptor &input_c_desc,
1761 const dnn::RnnStateTensorDescriptor &input_c_desc,
1779 const dnn::RnnStateTensorDescriptor &input_c_desc,
1800 const dnn::RnnStateTensorDescriptor &input_c_desc,
1825 const dnn::RnnStateTensorDescriptor &input_c_desc,
1850 const dnn::RnnStateTensorDescriptor &input_c_desc,
Dstream.cc4851 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnForward() argument
4868 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward()
4888 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnForward() argument
4904 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward()
4924 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnForward() argument
4941 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnForward()
4961 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnBackward() argument
4985 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnBackward()
5008 const dnn::RnnStateTensorDescriptor &input_c_desc, in ThenRnnBackward() argument
5031 input_c_desc, input_c_data, params, output_desc, output_data, in ThenRnnBackward()
[all …]