Searched defs:input_h_data (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1940 const DeviceMemory<T>& input_h_data, in ExtractAndCheckRnnForward()
2050 const DeviceMemory<T>& input_h_data, in DoRnnForwardImpl()
2165 const DeviceMemory<T>& input_h_data, in DoRnnBackwardImpl()
2375 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward()
2418 const DeviceMemory<float>& input_h_data, in DoRnnForward()
2460 const DeviceMemory<double>& input_h_data, in DoRnnForward()
2482 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnBackward()
2534 const DeviceMemory<float>& input_h_data, in DoRnnBackward()
2585 const DeviceMemory<double>& input_h_data, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1577 const DeviceMemory<T>& input_h_data, in ExtractAndCheckRnnForward()
1726 const DeviceMemory<T>& input_h_data, in DoRnnForwardImpl()
1878 const DeviceMemory<T>& input_h_data, in DoRnnBackwardImpl()
2128 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward()
2170 const DeviceMemory<float>& input_h_data, in DoRnnForward()
2211 const DeviceMemory<double>& input_h_data, in DoRnnForward()
2253 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnBackward()
2305 const DeviceMemory<float>& input_h_data, in DoRnnBackward()
2356 const DeviceMemory<double>& input_h_data, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 810 auto input_h_data = AsDeviceMemory<T>(input_h); in DoForward() local
873 auto input_h_data = AsDeviceMemory<T>(input_h); in DoBackward() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 4992 const DeviceMemory<Eigen::half> &input_h_data, in ThenRnnForward()
5029 const DeviceMemory<float> &input_h_data, in ThenRnnForward()
5065 const DeviceMemory<double> &input_h_data, in ThenRnnForward()
5102 const DeviceMemory<Eigen::half> &input_h_data, in ThenRnnBackward()
5149 const DeviceMemory<float> &input_h_data, in ThenRnnBackward()
5195 const DeviceMemory<double> &input_h_data, in ThenRnnBackward()
|