Searched refs:rnn_plan (Results 1 – 1 of 1) sorted by relevance
1234 PersistentRnnPlan rnn_plan, int num_layers, in CudnnRnnDescriptor() argument1244 rnn_plan_(std::move(rnn_plan)), in CudnnRnnDescriptor()1349 PersistentRnnPlan rnn_plan; in Create() local1357 rnn_plan = std::move(rnn_plan_wrapper).value(); in Create()1359 cudnnSetPersistentRNNPlan(rnn_desc.get(), rnn_plan.get())); in Create()1372 return CudnnRnnDescriptor(cudnn, std::move(rnn_desc), std::move(rnn_plan), in Create()