Searched refs:cudnn_input_h_desc (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2148 const CudnnRnnStateTensorDescriptor& cudnn_input_h_desc = in DoRnnForward() local
2161 cudnn_input_h_desc, input_h_data, cudnn_input_c_desc, input_c_data, in DoRnnForward()
2189 const CudnnRnnStateTensorDescriptor& cudnn_input_h_desc = in DoRnnForward() local
2202 cudnn_input_h_desc, input_h_data, cudnn_input_c_desc, input_c_data, in DoRnnForward()
2231 const CudnnRnnStateTensorDescriptor& cudnn_input_h_desc = in DoRnnForward() local
2244 cudnn_input_h_desc, input_h_data, cudnn_input_c_desc, input_c_data, in DoRnnForward()
2280 const CudnnRnnStateTensorDescriptor& cudnn_input_h_desc = in DoRnnBackward() local
2293 cudnn_input_h_desc, input_h_data, cudnn_input_c_desc, input_c_data, in DoRnnBackward()
2331 const CudnnRnnStateTensorDescriptor& cudnn_input_h_desc = in DoRnnBackward() local
2344 cudnn_input_h_desc, input_h_data, cudnn_input_c_desc, input_c_data, in DoRnnBackward()
[all …]
|