Home
last modified time | relevance | path

Searched refs:rnn_desc_s (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc964 auto rnn_desc_s = stream->parent()->createRnnDescriptor( in ExtractCudnnRNNParamsInfo() local
968 if (!rnn_desc_s.ok()) { in ExtractCudnnRNNParamsInfo()
969 return FromExecutorStatus(rnn_desc_s); in ExtractCudnnRNNParamsInfo()
971 *rnn_desc = rnn_desc_s.ConsumeValueOrDie(); in ExtractCudnnRNNParamsInfo()
984 auto rnn_desc_s = executor->createRnnDescriptor( in CreateRnnDescriptor() local
989 TF_RETURN_IF_ERROR(rnn_desc_s.status()); in CreateRnnDescriptor()
991 *rnn_desc = rnn_desc_s.ConsumeValueOrDie(); in CreateRnnDescriptor()