Searched refs:cudnn_probs_desc (Results 1 – 1 of 1) sorted by relevance
3899 const CudnnRnnStateTensorDescriptor& cudnn_probs_desc = in DoPrepareForCtcLoss() local3910 /*handle=*/cudnn.handle(), /*probsDesc=*/cudnn_probs_desc.handle(), in DoPrepareForCtcLoss()3925 /*handle=*/cudnn.handle(), /*probsDesc=*/cudnn_probs_desc.handle(), in DoPrepareForCtcLoss()3971 const CudnnRnnStateTensorDescriptor& cudnn_probs_desc = in DoCtcLoss() local3975 return DoCtcLossImpl(stream, cudnn_probs_desc, probs_data, labels_data, in DoCtcLoss()