Searched defs:input_h_data (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1416 const DeviceMemory<T>& input_h_data, in ExtractAndCheckRnnForward()
1511 const DeviceMemory<T>& input_h_data, in DoRnnForwardImpl()
1664 const DeviceMemory<T>& input_h_data, in DoRnnBackwardImpl()
1876 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward()
1918 const DeviceMemory<float>& input_h_data, in DoRnnForward()
1959 const DeviceMemory<double>& input_h_data, in DoRnnForward()
2001 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnBackward()
2053 const DeviceMemory<float>& input_h_data, in DoRnnBackward()
2104 const DeviceMemory<double>& input_h_data, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1896 const DeviceMemory<T>& input_h_data, in ExtractAndCheckRnnForward()
2006 const DeviceMemory<T>& input_h_data, in DoRnnForwardImpl()
2121 const DeviceMemory<T>& input_h_data, in DoRnnBackwardImpl()
2329 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward()
2372 const DeviceMemory<float>& input_h_data, in DoRnnForward()
2414 const DeviceMemory<double>& input_h_data, in DoRnnForward()
2436 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnBackward()
2488 const DeviceMemory<float>& input_h_data, in DoRnnBackward()
2539 const DeviceMemory<double>& input_h_data, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 755 auto input_h_data = AsDeviceMemory<T>(input_h); in DoForward() local
817 auto input_h_data = AsDeviceMemory<T>(input_h); in DoBackward() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 4850 const DeviceMemory<Eigen::half> &input_h_data, in ThenRnnForward()
4887 const DeviceMemory<float> &input_h_data, in ThenRnnForward()
4923 const DeviceMemory<double> &input_h_data, in ThenRnnForward()
4960 const DeviceMemory<Eigen::half> &input_h_data, in ThenRnnBackward()
5007 const DeviceMemory<float> &input_h_data, in ThenRnnBackward()
5053 const DeviceMemory<double> &input_h_data, in ThenRnnBackward()
|