Searched refs:rnn_desc_ (Results 1 – 2 of 2) sorted by relevance
1671 const MIOpenRnnDescriptor* rnn_desc_; member in stream_executor::gpu::MIOpenRnnParamsDescriptor1688 : rnn_desc_(nullptr), in MIOpenRnnDescriptor()1697 auto status = wrap::miopenCreateRNNDescriptor(&rnn_desc_); in MIOpenRnnDescriptor()1700 rnn_desc_ /*rnnDesc*/, hidden_size /*hiddenSize*/, in MIOpenRnnDescriptor()1715 if (rnn_desc_) { in ~MIOpenRnnDescriptor()1716 auto status = wrap::miopenDestroyRNNDescriptor(rnn_desc_); in ~MIOpenRnnDescriptor()1722 return rnn_desc_; in handle()1748 miopenRNNDescriptor_t rnn_desc_; member in stream_executor::gpu::MIOpenRnnDescriptor1766 auto rnn_mode = rnn_desc_->rnn_mode(); in GetRegionCountPerLayer()2235 : handle_(nullptr), rnn_desc_(&rnn_desc), params_size_in_bytes_(0) { in MIOpenRnnParamsDescriptor()
1013 : rnn_desc_(std::move(rnn_desc)), in CudnnRnnDescriptor()1107 cudnnRNNDescriptor_t handle() const { return rnn_desc_.get(); } in handle()1134 gpu::RnnDescriptor rnn_desc_; member in stream_executor::gpu::CudnnRnnDescriptor