Searched defs:output_h_data (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1422 const DeviceMemory<T>& output_h_data, in ExtractAndCheckRnnForward()
1517 DeviceMemory<T>* output_h_data, in DoRnnForwardImpl()
1670 const DeviceMemory<T>& output_h_data, in DoRnnBackwardImpl()
1883 DeviceMemory<Eigen::half>* output_h_data, in DoRnnForward()
1924 DeviceMemory<float>* output_h_data, in DoRnnForward()
1966 DeviceMemory<double>* output_h_data, in DoRnnForward()
2008 const DeviceMemory<Eigen::half>& output_h_data, in DoRnnBackward()
2059 const DeviceMemory<float>& output_h_data, in DoRnnBackward()
2111 const DeviceMemory<double>& output_h_data, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1902 const DeviceMemory<T>& output_h_data, in ExtractAndCheckRnnForward()
2012 DeviceMemory<T>* output_h_data, in DoRnnForwardImpl()
2127 const DeviceMemory<T>& output_h_data, in DoRnnBackwardImpl()
2336 DeviceMemory<Eigen::half>* output_h_data, in DoRnnForward()
2378 DeviceMemory<float>* output_h_data, in DoRnnForward()
2421 DeviceMemory<double>* output_h_data, in DoRnnForward()
2443 const DeviceMemory<Eigen::half>& output_h_data, in DoRnnBackward()
2494 const DeviceMemory<float>& output_h_data, in DoRnnBackward()
2546 const DeviceMemory<double>& output_h_data, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 763 auto output_h_data = AsDeviceMemory<T>(output_h); in DoForward() local
824 auto output_h_data = AsDeviceMemory<T>(output_h); in DoBackward() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 4857 DeviceMemory<Eigen::half> *output_h_data, in ThenRnnForward()
4893 DeviceMemory<float> *output_h_data, in ThenRnnForward()
4930 DeviceMemory<double> *output_h_data, in ThenRnnForward()
4967 const DeviceMemory<Eigen::half> &output_h_data, in ThenRnnBackward()
5013 const DeviceMemory<float> &output_h_data, in ThenRnnBackward()
5060 const DeviceMemory<double> &output_h_data, in ThenRnnBackward()
|