Searched refs:cudnn_output_h_desc (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2154 const CudnnRnnStateTensorDescriptor& cudnn_output_h_desc = in DoRnnForward() local
2162 params, cudnn_output_desc, output_data, cudnn_output_h_desc, in DoRnnForward()
2195 const CudnnRnnStateTensorDescriptor& cudnn_output_h_desc = in DoRnnForward() local
2203 params, cudnn_output_desc, output_data, cudnn_output_h_desc, in DoRnnForward()
2237 const CudnnRnnStateTensorDescriptor& cudnn_output_h_desc = in DoRnnForward() local
2245 params, cudnn_output_desc, output_data, cudnn_output_h_desc, in DoRnnForward()
2286 const CudnnRnnStateTensorDescriptor& cudnn_output_h_desc = in DoRnnBackward() local
2294 params, cudnn_output_desc, output_data, cudnn_output_h_desc, in DoRnnBackward()
2337 const CudnnRnnStateTensorDescriptor& cudnn_output_h_desc = in DoRnnBackward() local
2345 params, cudnn_output_desc, output_data, cudnn_output_h_desc, in DoRnnBackward()
[all …]
|