Searched refs:cudnn_probs_desc (Results 1 – 1 of 1) sorted by relevance
5850 const CudnnRnnStateTensorDescriptor& cudnn_probs_desc = in DoPrepareForCtcLoss() local5861 /*handle=*/cudnn.handle(), /*probsDesc=*/cudnn_probs_desc.handle(), in DoPrepareForCtcLoss()5876 /*handle=*/cudnn.handle(), /*probsDesc=*/cudnn_probs_desc.handle(), in DoPrepareForCtcLoss()5922 const CudnnRnnStateTensorDescriptor& cudnn_probs_desc = in DoCtcLoss() local5926 return DoCtcLossImpl(stream, cudnn_probs_desc, probs_data, labels_data, in DoCtcLoss()