Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1415 const CudnnRnnStateTensorDescriptor& input_h_desc, in ExtractAndCheckRnnForward() argument
1436 if (!(input_h_desc.num_layers() == in ExtractAndCheckRnnForward()
1438 input_h_desc.batch_size() == model_dims.batch_size && in ExtractAndCheckRnnForward()
1439 input_h_desc.data_size() == model_dims.hidden_size)) { in ExtractAndCheckRnnForward()
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()
1453 if (!(input_h_desc.num_layers() == output_h_desc.num_layers() && in ExtractAndCheckRnnForward()
1454 input_h_desc.batch_size() == output_h_desc.batch_size() && in ExtractAndCheckRnnForward()
1455 input_h_desc.data_size() == output_h_desc.data_size())) { in ExtractAndCheckRnnForward()
[all …]
Dcuda_dnn.h76 const dnn::RnnStateTensorDescriptor& input_h_desc,
94 const dnn::RnnStateTensorDescriptor& input_h_desc,
112 const dnn::RnnStateTensorDescriptor& input_h_desc,
130 const dnn::RnnStateTensorDescriptor& input_h_desc,
155 const dnn::RnnStateTensorDescriptor& input_h_desc,
180 const dnn::RnnStateTensorDescriptor& input_h_desc,
610 const CudnnRnnStateTensorDescriptor& input_h_desc,
629 const CudnnRnnStateTensorDescriptor& input_h_desc,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1895 const MIOpenRnnStateTensorDescriptor& input_h_desc, in ExtractAndCheckRnnForward() argument
1915 if (!(input_h_desc.num_layers() == in ExtractAndCheckRnnForward()
1917 input_h_desc.batch_size() == model_dims->batch_size && in ExtractAndCheckRnnForward()
1918 input_h_desc.data_size() == model_dims->hidden_size)) { in ExtractAndCheckRnnForward()
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()
1935 if (!(input_h_desc.num_layers() == output_h_desc.num_layers() && in ExtractAndCheckRnnForward()
1936 input_h_desc.batch_size() == output_h_desc.batch_size() && in ExtractAndCheckRnnForward()
1937 input_h_desc.data_size() == output_h_desc.data_size())) { in ExtractAndCheckRnnForward()
[all …]
Drocm_dnn.h67 const dnn::RnnStateTensorDescriptor& input_h_desc,
85 const dnn::RnnStateTensorDescriptor& input_h_desc,
103 const dnn::RnnStateTensorDescriptor& input_h_desc,
121 const dnn::RnnStateTensorDescriptor& input_h_desc,
146 const dnn::RnnStateTensorDescriptor& input_h_desc,
171 const dnn::RnnStateTensorDescriptor& input_h_desc,
666 const MIOpenRnnStateTensorDescriptor& input_h_desc,
683 const MIOpenRnnStateTensorDescriptor& input_h_desc,
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2119 const dnn::RnnStateTensorDescriptor& input_h_desc,
2140 const dnn::RnnStateTensorDescriptor& input_h_desc,
2161 const dnn::RnnStateTensorDescriptor& input_h_desc,
2223 const dnn::RnnStateTensorDescriptor& input_h_desc,
2251 const dnn::RnnStateTensorDescriptor& input_h_desc,
2279 const dnn::RnnStateTensorDescriptor& input_h_desc,
Dstream.h1740 const dnn::RnnStateTensorDescriptor &input_h_desc,
1759 const dnn::RnnStateTensorDescriptor &input_h_desc,
1777 const dnn::RnnStateTensorDescriptor &input_h_desc,
1798 const dnn::RnnStateTensorDescriptor &input_h_desc,
1823 const dnn::RnnStateTensorDescriptor &input_h_desc,
1848 const dnn::RnnStateTensorDescriptor &input_h_desc,
Dstream.cc4849 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnForward() argument
4867 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnForward()
4886 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnForward() argument
4903 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnForward()
4922 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnForward() argument
4940 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnForward()
4959 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnBackward() argument
4984 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnBackward()
5006 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnBackward() argument
5030 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnBackward()
[all …]