Searched refs:CudnnRNNForwardOp (Results 1 – 1 of 1) sorted by relevance
97 class CudnnRNNForwardOp;1252 class CudnnRNNForwardOp<GPUDevice, T> : public CudnnRNNKernelCommon { class1254 explicit CudnnRNNForwardOp(OpKernelConstruction* context) in CudnnRNNForwardOp() function in tensorflow::CudnnRNNForwardOp1392 CudnnRNNForwardOp<GPUDevice, T>);1401 : public CudnnRNNForwardOp<GPUDevice, T> {1403 using CudnnRNNForwardOp<GPUDevice, T>::is_training;1411 : CudnnRNNForwardOp<GPUDevice, T>(context) {} in CudnnRNNForwardOpV2()1415 CudnnRNNForwardOp<GPUDevice, T>::ComputeAndReturnAlgorithm( in Compute()1600 : public CudnnRNNForwardOp<GPUDevice, T> {1602 using CudnnRNNForwardOp<GPUDevice, T>::is_training;[all …]