Searched refs:rnn_plan (Results 1 – 1 of 1) sorted by relevance
1059 PersistentRnnPlan rnn_plan, int num_layers, in CudnnRnnDescriptor() argument1069 rnn_plan_(std::move(rnn_plan)), in CudnnRnnDescriptor()1174 PersistentRnnPlan rnn_plan; in Create() local1182 rnn_plan = rnn_plan_wrapper.ConsumeValueOrDie(); in Create()1184 cudnnSetPersistentRNNPlan(rnn_desc.get(), rnn_plan.get())); in Create()1194 return CudnnRnnDescriptor(cudnn, std::move(rnn_desc), std::move(rnn_plan), in Create()