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.cc1601 const CudnnRnnStateTensorDescriptor& input_h_desc, in ExtractAndCheckRnnForward() argument
1623 if (!(input_h_desc.num_layers() == in ExtractAndCheckRnnForward()
1625 input_h_desc.batch_size() == model_dims.batch_size && in ExtractAndCheckRnnForward()
1626 input_h_desc.data_size() == model_dims.hidden_size)) { in ExtractAndCheckRnnForward()
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()
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()
[all …]
Dcuda_dnn.h77 const dnn::RnnStateTensorDescriptor& input_h_desc,
95 const dnn::RnnStateTensorDescriptor& input_h_desc,
113 const dnn::RnnStateTensorDescriptor& input_h_desc,
131 const dnn::RnnStateTensorDescriptor& input_h_desc,
156 const dnn::RnnStateTensorDescriptor& input_h_desc,
181 const dnn::RnnStateTensorDescriptor& input_h_desc,
644 const CudnnRnnStateTensorDescriptor& input_h_desc,
663 const CudnnRnnStateTensorDescriptor& input_h_desc,
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1976 const MIOpenRnnStateTensorDescriptor& input_h_desc, in ExtractAndCheckRnnForward() argument
1996 if (!(input_h_desc.num_layers() == in ExtractAndCheckRnnForward()
1998 input_h_desc.batch_size() == model_dims->batch_size && in ExtractAndCheckRnnForward()
1999 input_h_desc.data_size() == model_dims->hidden_size)) { in ExtractAndCheckRnnForward()
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()
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()
[all …]
Drocm_dnn.h104 const dnn::RnnStateTensorDescriptor& input_h_desc,
122 const dnn::RnnStateTensorDescriptor& input_h_desc,
140 const dnn::RnnStateTensorDescriptor& input_h_desc,
158 const dnn::RnnStateTensorDescriptor& input_h_desc,
183 const dnn::RnnStateTensorDescriptor& input_h_desc,
208 const dnn::RnnStateTensorDescriptor& input_h_desc,
741 const MIOpenRnnStateTensorDescriptor& input_h_desc,
758 const MIOpenRnnStateTensorDescriptor& input_h_desc,
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2188 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward() argument
2209 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward() argument
2230 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnForward() argument
2292 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward() argument
2320 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward() argument
2348 const dnn::RnnStateTensorDescriptor& input_h_desc, in DoRnnBackward() argument
Dstream.h1782 const dnn::RnnStateTensorDescriptor &input_h_desc,
1801 const dnn::RnnStateTensorDescriptor &input_h_desc,
1819 const dnn::RnnStateTensorDescriptor &input_h_desc,
1840 const dnn::RnnStateTensorDescriptor &input_h_desc,
1865 const dnn::RnnStateTensorDescriptor &input_h_desc,
1890 const dnn::RnnStateTensorDescriptor &input_h_desc,
Dstream.cc4542 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnForward() argument
4559 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnForward()
4576 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnForward() argument
4592 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnForward()
4609 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnForward() argument
4626 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnForward()
4643 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnBackward() argument
4667 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnBackward()
4688 const dnn::RnnStateTensorDescriptor &input_h_desc, in ThenRnnBackward() argument
4711 this, rnn_desc, input_desc, input_data, input_h_desc, input_h_data, in ThenRnnBackward()
[all …]