Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h83 const dnn::RnnStateTensorDescriptor& output_h_desc,
101 const dnn::RnnStateTensorDescriptor& output_h_desc,
119 const dnn::RnnStateTensorDescriptor& output_h_desc,
137 const dnn::RnnStateTensorDescriptor& output_h_desc,
162 const dnn::RnnStateTensorDescriptor& output_h_desc,
187 const dnn::RnnStateTensorDescriptor& output_h_desc,
616 const CudnnRnnStateTensorDescriptor& output_h_desc,
635 const CudnnRnnStateTensorDescriptor& output_h_desc,
Dcuda_dnn.cc1421 const CudnnRnnStateTensorDescriptor& output_h_desc, in ExtractAndCheckRnnForward() argument
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()
1516 const CudnnRnnStateTensorDescriptor& output_h_desc, in DoRnnForwardImpl() argument
1528 output_h_desc, *output_h_data, output_c_desc, *output_c_data)); in DoRnnForwardImpl()
1577 /*hyDesc=*/output_h_desc.handle(), /*hy=*/output_h_data->opaque(), in DoRnnForwardImpl()
1597 /*y=*/output_data->opaque(), /*hyDesc=*/output_h_desc.handle(), in DoRnnForwardImpl()
1614 /*hyDesc=*/output_h_desc.handle(), /*hy=*/output_h_data->opaque(), in DoRnnForwardImpl()
1636 /*y=*/output_data->opaque(), /*hyDesc=*/output_h_desc.handle(), in DoRnnForwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h74 const dnn::RnnStateTensorDescriptor& output_h_desc,
92 const dnn::RnnStateTensorDescriptor& output_h_desc,
110 const dnn::RnnStateTensorDescriptor& output_h_desc,
128 const dnn::RnnStateTensorDescriptor& output_h_desc,
153 const dnn::RnnStateTensorDescriptor& output_h_desc,
178 const dnn::RnnStateTensorDescriptor& output_h_desc,
673 const MIOpenRnnStateTensorDescriptor& output_h_desc,
690 const MIOpenRnnStateTensorDescriptor& output_h_desc,
Drocm_dnn.cc1901 const MIOpenRnnStateTensorDescriptor& output_h_desc, in ExtractAndCheckRnnForward() argument
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()
2011 const MIOpenRnnStateTensorDescriptor& output_h_desc, in DoRnnForwardImpl() argument
2022 output_h_desc, *output_h_data, output_c_desc, *output_c_data, in DoRnnForwardImpl()
2082 output_data->opaque() /*y*/, output_h_desc.handle() /*hyDesc*/, in DoRnnForwardImpl()
2100 output_data->opaque() /*y*/, output_h_desc.handle() /*hyDesc*/, in DoRnnForwardImpl()
2126 const MIOpenRnnStateTensorDescriptor& output_h_desc, in DoRnnBackwardImpl() argument
2144 output_h_desc, output_h_data, output_c_desc, output_c_data, &model_dims); in DoRnnBackwardImpl()
[all …]
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2126 const dnn::RnnStateTensorDescriptor& output_h_desc,
2147 const dnn::RnnStateTensorDescriptor& output_h_desc,
2168 const dnn::RnnStateTensorDescriptor& output_h_desc,
2230 const dnn::RnnStateTensorDescriptor& output_h_desc,
2258 const dnn::RnnStateTensorDescriptor& output_h_desc,
2286 const dnn::RnnStateTensorDescriptor& output_h_desc,
Dstream.h1747 const dnn::RnnStateTensorDescriptor &output_h_desc,
1766 const dnn::RnnStateTensorDescriptor &output_h_desc,
1784 const dnn::RnnStateTensorDescriptor &output_h_desc,
1805 const dnn::RnnStateTensorDescriptor &output_h_desc,
1830 const dnn::RnnStateTensorDescriptor &output_h_desc,
1855 const dnn::RnnStateTensorDescriptor &output_h_desc,
Dstream.cc4856 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnForward() argument
4869 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward()
4892 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnForward() argument
4905 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward()
4929 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnForward() argument
4942 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward()
4966 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnBackward() argument
4986 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward()
5012 const dnn::RnnStateTensorDescriptor &output_h_desc, in ThenRnnBackward() argument
5032 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward()
[all …]