Searched refs:MIOpenRnnStateTensorDescriptor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1967 class MIOpenRnnStateTensorDescriptor class
1970 MIOpenRnnStateTensorDescriptor(int num_layers, int batch_size, int data_size, in MIOpenRnnStateTensorDescriptor() function in stream_executor::gpu::MIOpenRnnStateTensorDescriptor
1986 ~MIOpenRnnStateTensorDescriptor() override { in ~MIOpenRnnStateTensorDescriptor()
2008 SE_DISALLOW_COPY_AND_ASSIGN(MIOpenRnnStateTensorDescriptor);
2027 const MIOpenRnnStateTensorDescriptor& input_h_desc, in ExtractAndCheckRnnForward()
2029 const MIOpenRnnStateTensorDescriptor& input_c_desc, in ExtractAndCheckRnnForward()
2033 const MIOpenRnnStateTensorDescriptor& output_h_desc, in ExtractAndCheckRnnForward()
2035 const MIOpenRnnStateTensorDescriptor& output_c_desc, in ExtractAndCheckRnnForward()
2137 const MIOpenRnnStateTensorDescriptor& input_h_desc, in DoRnnForwardImpl()
2139 const MIOpenRnnStateTensorDescriptor& input_c_desc, in DoRnnForwardImpl()
[all …]
|
| D | rocm_dnn.h | 36 class MIOpenRnnStateTensorDescriptor; variable
679 const MIOpenRnnStateTensorDescriptor& input_h_desc,
681 const MIOpenRnnStateTensorDescriptor& input_c_desc,
686 const MIOpenRnnStateTensorDescriptor& output_h_desc,
688 const MIOpenRnnStateTensorDescriptor& output_c_desc,
696 const MIOpenRnnStateTensorDescriptor& input_h_desc,
698 const MIOpenRnnStateTensorDescriptor& input_c_desc,
703 const MIOpenRnnStateTensorDescriptor& output_h_desc,
705 const MIOpenRnnStateTensorDescriptor& output_c_desc,
784 Stream* stream, const MIOpenRnnStateTensorDescriptor& probs_desc,
[all …]
|