Searched refs:cudnn_output_c_desc (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2156 const CudnnRnnStateTensorDescriptor& cudnn_output_c_desc = in DoRnnForward() local
2163 output_h_data, cudnn_output_c_desc, output_c_data, is_training, in DoRnnForward()
2197 const CudnnRnnStateTensorDescriptor& cudnn_output_c_desc = in DoRnnForward() local
2204 output_h_data, cudnn_output_c_desc, output_c_data, is_training, in DoRnnForward()
2239 const CudnnRnnStateTensorDescriptor& cudnn_output_c_desc = in DoRnnForward() local
2246 output_h_data, cudnn_output_c_desc, output_c_data, is_training, in DoRnnForward()
2288 const CudnnRnnStateTensorDescriptor& cudnn_output_c_desc = in DoRnnBackward() local
2295 output_h_data, cudnn_output_c_desc, output_c_data, in DoRnnBackward()
2339 const CudnnRnnStateTensorDescriptor& cudnn_output_c_desc = in DoRnnBackward() local
2346 output_h_data, cudnn_output_c_desc, output_c_data, in DoRnnBackward()
[all …]
|