Searched refs:MIOpenRnnStateTensorDescriptor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1835 class MIOpenRnnStateTensorDescriptor class
1838 MIOpenRnnStateTensorDescriptor(int num_layers, int batch_size, int data_size, in MIOpenRnnStateTensorDescriptor() function in stream_executor::gpu::MIOpenRnnStateTensorDescriptor
1854 ~MIOpenRnnStateTensorDescriptor() override { in ~MIOpenRnnStateTensorDescriptor()
1876 SE_DISALLOW_COPY_AND_ASSIGN(MIOpenRnnStateTensorDescriptor);
1895 const MIOpenRnnStateTensorDescriptor& input_h_desc, in ExtractAndCheckRnnForward()
1897 const MIOpenRnnStateTensorDescriptor& input_c_desc, in ExtractAndCheckRnnForward()
1901 const MIOpenRnnStateTensorDescriptor& output_h_desc, in ExtractAndCheckRnnForward()
1903 const MIOpenRnnStateTensorDescriptor& output_c_desc, in ExtractAndCheckRnnForward()
2005 const MIOpenRnnStateTensorDescriptor& input_h_desc, in DoRnnForwardImpl()
2007 const MIOpenRnnStateTensorDescriptor& input_c_desc, in DoRnnForwardImpl()
[all …]
|
| D | rocm_dnn.h | 35 class MIOpenRnnStateTensorDescriptor; variable
666 const MIOpenRnnStateTensorDescriptor& input_h_desc,
668 const MIOpenRnnStateTensorDescriptor& input_c_desc,
673 const MIOpenRnnStateTensorDescriptor& output_h_desc,
675 const MIOpenRnnStateTensorDescriptor& output_c_desc,
683 const MIOpenRnnStateTensorDescriptor& input_h_desc,
685 const MIOpenRnnStateTensorDescriptor& input_c_desc,
690 const MIOpenRnnStateTensorDescriptor& output_h_desc,
692 const MIOpenRnnStateTensorDescriptor& output_c_desc,
|