Searched refs:rnn_desc_ (Results 1 – 2 of 2) sorted by relevance
1752 const MIOpenRnnDescriptor* rnn_desc_; member in stream_executor::gpu::MIOpenRnnParamsDescriptor1769 : rnn_desc_(nullptr), in MIOpenRnnDescriptor()1778 auto status = wrap::miopenCreateRNNDescriptor(&rnn_desc_); in MIOpenRnnDescriptor()1781 rnn_desc_ /*rnnDesc*/, hidden_size /*hiddenSize*/, in MIOpenRnnDescriptor()1796 if (rnn_desc_) { in ~MIOpenRnnDescriptor()1797 auto status = wrap::miopenDestroyRNNDescriptor(rnn_desc_); in ~MIOpenRnnDescriptor()1803 return rnn_desc_; in handle()1829 miopenRNNDescriptor_t rnn_desc_; member in stream_executor::gpu::MIOpenRnnDescriptor1847 auto rnn_mode = rnn_desc_->rnn_mode(); in GetRegionCountPerLayer()2317 : handle_(nullptr), rnn_desc_(&rnn_desc), params_size_in_bytes_(0) { in MIOpenRnnParamsDescriptor()
1068 : rnn_desc_(std::move(rnn_desc)), in CudnnRnnDescriptor()1201 cudnnRNNDescriptor_t handle() const { return rnn_desc_.get(); } in handle()1229 gpu::RnnDescriptor rnn_desc_; member in stream_executor::gpu::CudnnRnnDescriptor