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.h86 const dnn::RnnStateTensorDescriptor& output_c_desc,
104 const dnn::RnnStateTensorDescriptor& output_c_desc,
122 const dnn::RnnStateTensorDescriptor& output_c_desc,
140 const dnn::RnnStateTensorDescriptor& output_c_desc,
165 const dnn::RnnStateTensorDescriptor& output_c_desc,
190 const dnn::RnnStateTensorDescriptor& output_c_desc,
652 const CudnnRnnStateTensorDescriptor& output_c_desc,
671 const CudnnRnnStateTensorDescriptor& output_c_desc,
Dcuda_dnn.cc1609 const CudnnRnnStateTensorDescriptor& output_c_desc, in ExtractAndCheckRnnForward() argument
1648 if (!(input_h_desc.num_layers() == output_c_desc.num_layers() && in ExtractAndCheckRnnForward()
1649 input_h_desc.batch_size() == output_c_desc.batch_size() && in ExtractAndCheckRnnForward()
1650 input_h_desc.data_size() <= output_c_desc.data_size())) { in ExtractAndCheckRnnForward()
1758 const CudnnRnnStateTensorDescriptor& output_c_desc, in DoRnnForwardImpl() argument
1768 output_h_desc, *output_h_data, output_c_desc, *output_c_data)); in DoRnnForwardImpl()
1816 /*cyDesc=*/output_c_desc.handle(), /*cy=*/output_c_data->opaque(), in DoRnnForwardImpl()
1831 /*hy=*/output_h_data->opaque(), /*cyDesc=*/output_c_desc.handle(), in DoRnnForwardImpl()
1847 /*cyDesc=*/output_c_desc.handle(), /*cy=*/output_c_data->opaque(), in DoRnnForwardImpl()
1864 /*hy=*/output_h_data->opaque(), /*cyDesc=*/output_c_desc.handle(), in DoRnnForwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h113 const dnn::RnnStateTensorDescriptor& output_c_desc,
131 const dnn::RnnStateTensorDescriptor& output_c_desc,
149 const dnn::RnnStateTensorDescriptor& output_c_desc,
167 const dnn::RnnStateTensorDescriptor& output_c_desc,
192 const dnn::RnnStateTensorDescriptor& output_c_desc,
217 const dnn::RnnStateTensorDescriptor& output_c_desc,
750 const MIOpenRnnStateTensorDescriptor& output_c_desc,
767 const MIOpenRnnStateTensorDescriptor& output_c_desc,
Drocm_dnn.cc1984 const MIOpenRnnStateTensorDescriptor& output_c_desc, in ExtractAndCheckRnnForward() argument
2022 if (!(input_h_desc.num_layers() == output_c_desc.num_layers() && in ExtractAndCheckRnnForward()
2023 input_h_desc.batch_size() == output_c_desc.batch_size() && in ExtractAndCheckRnnForward()
2024 input_h_desc.data_size() == output_c_desc.data_size())) { in ExtractAndCheckRnnForward()
2094 const MIOpenRnnStateTensorDescriptor& output_c_desc, in DoRnnForwardImpl() argument
2103 output_h_desc, *output_h_data, output_c_desc, *output_c_data, in DoRnnForwardImpl()
2164 output_h_data->opaque() /*hy*/, output_c_desc.handle() /*cyDesc*/, in DoRnnForwardImpl()
2182 output_h_data->opaque() /*hy*/, output_c_desc.handle() /*cyDesc*/, in DoRnnForwardImpl()
2209 const MIOpenRnnStateTensorDescriptor& output_c_desc, in DoRnnBackwardImpl() argument
2225 output_h_desc, output_h_data, output_c_desc, output_c_data, &model_dims); in DoRnnBackwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2197 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward() argument
2218 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward() argument
2239 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnForward() argument
2301 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward() argument
2329 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward() argument
2357 const dnn::RnnStateTensorDescriptor& output_c_desc, in DoRnnBackward() argument
Dstream.h1791 const dnn::RnnStateTensorDescriptor &output_c_desc,
1810 const dnn::RnnStateTensorDescriptor &output_c_desc,
1828 const dnn::RnnStateTensorDescriptor &output_c_desc,
1849 const dnn::RnnStateTensorDescriptor &output_c_desc,
1874 const dnn::RnnStateTensorDescriptor &output_c_desc,
1899 const dnn::RnnStateTensorDescriptor &output_c_desc,
Dstream.cc4551 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward() argument
4561 output_h_desc, output_h_data, output_c_desc, output_c_data, is_training, in ThenRnnForward()
4584 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward() argument
4594 output_h_desc, output_h_data, output_c_desc, output_c_data, is_training, in ThenRnnForward()
4618 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnForward() argument
4628 output_h_desc, output_h_data, output_c_desc, output_c_data, is_training, in ThenRnnForward()
4652 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnBackward() argument
4669 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward()
4696 const dnn::RnnStateTensorDescriptor &output_c_desc, in ThenRnnBackward() argument
4713 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward()
[all …]