Searched refs:CreateRnnDescriptor (Results 1 – 2 of 2) sorted by relevance
976 Status CreateRnnDescriptor(OpKernelContext* context, in CreateRnnDescriptor() function in tensorflow::CudnnRNNKernelCommon1014 CreateRnnDescriptor<T>(context, model_shapes, input_mode, algo_config, in GetCachedRnnDescriptor()1404 using CudnnRNNKernelCommon::CreateRnnDescriptor;1525 if (!this->template CreateRnnDescriptor<T>( in MaybeAutoTune()1603 using CudnnRNNKernelCommon::CreateRnnDescriptor;
477 RnnDescriptor CreateRnnDescriptor() { in CreateRnnDescriptor() function1043 gpu::RnnDescriptor rnn_desc = CreateRnnDescriptor(); in Create()