Searched refs:cudnn_grads_desc (Results 1 – 1 of 1) sorted by relevance
3901 const CudnnRnnStateTensorDescriptor& cudnn_grads_desc = in DoPrepareForCtcLoss() local3911 /*gradientsDesc=*/cudnn_grads_desc.handle(), in DoPrepareForCtcLoss()3926 /*gradientsDesc=*/cudnn_grads_desc.handle(), in DoPrepareForCtcLoss()3973 const CudnnRnnStateTensorDescriptor& cudnn_grads_desc = in DoCtcLoss() local3977 cudnn_grads_desc, grads_data, cudnn_ctc_loss_desc, in DoCtcLoss()