Searched defs:input_h_data (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1428 const DeviceMemory<T>& input_h_data, in ExtractAndCheckRnnForward()
1538 const DeviceMemory<T>& input_h_data, in DoRnnForwardImpl()
1655 const DeviceMemory<T>& input_h_data, in DoRnnBackwardImpl()
1833 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward()
1876 const DeviceMemory<float>& input_h_data, in DoRnnForward()
1918 const DeviceMemory<double>& input_h_data, in DoRnnForward()
1961 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnBackward()
2014 const DeviceMemory<float>& input_h_data, in DoRnnBackward()
2066 const DeviceMemory<double>& input_h_data, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/ |
| D | cudnn_rnn_ops.cc | 861 auto input_h_data = AsDeviceMemory<T>(input_h); in Compute() local
1061 auto input_h_data = AsDeviceMemory<T>(input_h); in Compute() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 4687 const DeviceMemory<Eigen::half> &input_h_data, in ThenRnnForward()
4720 const DeviceMemory<float> &input_h_data, in ThenRnnForward()
4752 const DeviceMemory<double> &input_h_data, in ThenRnnForward()
4785 const DeviceMemory<Eigen::half> &input_h_data, in ThenRnnBackward()
4827 const DeviceMemory<float> &input_h_data, in ThenRnnBackward()
4868 const DeviceMemory<double> &input_h_data, in ThenRnnBackward()
|