Searched defs:output_h_data (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1434 const DeviceMemory<T>& output_h_data, in ExtractAndCheckRnnForward()
1544 DeviceMemory<T>* output_h_data, in DoRnnForwardImpl()
1661 const DeviceMemory<T>& output_h_data, in DoRnnBackwardImpl()
1840 DeviceMemory<Eigen::half>* output_h_data, in DoRnnForward()
1882 DeviceMemory<float>* output_h_data, in DoRnnForward()
1925 DeviceMemory<double>* output_h_data, in DoRnnForward()
1968 const DeviceMemory<Eigen::half>& output_h_data, in DoRnnBackward()
2020 const DeviceMemory<float>& output_h_data, in DoRnnBackward()
2073 const DeviceMemory<double>& output_h_data, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/ |
| D | cudnn_rnn_ops.cc | 868 auto output_h_data = AsDeviceMemory<T>(output_h); in Compute() local
1068 auto output_h_data = AsDeviceMemory<T>(output_h); in Compute() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 4694 DeviceMemory<Eigen::half> *output_h_data, in ThenRnnForward()
4726 DeviceMemory<float> *output_h_data, in ThenRnnForward()
4759 DeviceMemory<double> *output_h_data, in ThenRnnForward()
4792 const DeviceMemory<Eigen::half> &output_h_data, in ThenRnnBackward()
4833 const DeviceMemory<float> &output_h_data, in ThenRnnBackward()
4875 const DeviceMemory<double> &output_h_data, in ThenRnnBackward()
|