Searched refs:MIOpenRnnSequenceTensorDescriptor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1780 class MIOpenRnnSequenceTensorDescriptor class
1783 MIOpenRnnSequenceTensorDescriptor(int seq_length, int batch_size, in MIOpenRnnSequenceTensorDescriptor() function in stream_executor::gpu::MIOpenRnnSequenceTensorDescriptor
1808 ~MIOpenRnnSequenceTensorDescriptor() override { in ~MIOpenRnnSequenceTensorDescriptor()
1832 SE_DISALLOW_COPY_AND_ASSIGN(MIOpenRnnSequenceTensorDescriptor);
1893 const MIOpenRnnSequenceTensorDescriptor& input_desc, in ExtractAndCheckRnnForward()
1899 const MIOpenRnnSequenceTensorDescriptor& output_desc, in ExtractAndCheckRnnForward()
1953 const MIOpenRnnSequenceTensorDescriptor& input_desc) { in CheckRNNParameterSize()
1969 const MIOpenRnnSequenceTensorDescriptor& input_desc, in CreateRnnWorkspace()
2003 const MIOpenRnnSequenceTensorDescriptor& input_desc, in DoRnnForwardImpl()
2009 const MIOpenRnnSequenceTensorDescriptor& output_desc, in DoRnnForwardImpl()
[all …]
|
| D | rocm_dnn.h | 34 class MIOpenRnnSequenceTensorDescriptor; variable
664 const MIOpenRnnSequenceTensorDescriptor& input_desc,
671 const MIOpenRnnSequenceTensorDescriptor& output_desc,
681 const MIOpenRnnSequenceTensorDescriptor& input_desc,
688 const MIOpenRnnSequenceTensorDescriptor& output_desc,
|