Searched refs:rnn_plan (Results 1 – 1 of 1) sorted by relevance
1004 PersistentRnnPlan rnn_plan, int num_layers, in CudnnRnnDescriptor() argument1014 rnn_plan_(std::move(rnn_plan)), in CudnnRnnDescriptor()1063 PersistentRnnPlan rnn_plan; in Create() local1071 rnn_plan = rnn_plan_wrapper.ConsumeValueOrDie(); in Create()1073 cudnnSetPersistentRNNPlan(rnn_desc.get(), rnn_plan.get())); in Create()1100 return CudnnRnnDescriptor(cudnn, std::move(rnn_desc), std::move(rnn_plan), in Create()