Home
last modified time | relevance | path

Searched defs:input_h_data (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2092 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward()
2114 const DeviceMemory<float>& input_h_data, in DoRnnForward()
2136 const DeviceMemory<double>& input_h_data, in DoRnnForward()
2199 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnBackward()
2228 const DeviceMemory<float>& input_h_data, in DoRnnBackward()
2257 const DeviceMemory<double>& input_h_data, in DoRnnBackward()
Dstream.cc3987 const DeviceMemory<Eigen::half> &input_h_data, in ThenRnnForward()
4023 const DeviceMemory<float> &input_h_data, in ThenRnnForward()
4058 const DeviceMemory<double> &input_h_data, in ThenRnnForward()
4094 const DeviceMemory<Eigen::half> &input_h_data, in ThenRnnBackward()
4140 const DeviceMemory<float> &input_h_data, in ThenRnnBackward()
4185 const DeviceMemory<double> &input_h_data, in ThenRnnBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2028 const DeviceMemory<T>& input_h_data, in ExtractAndCheckRnnForward()
2138 const DeviceMemory<T>& input_h_data, in DoRnnForwardImpl()
2253 const DeviceMemory<T>& input_h_data, in DoRnnBackwardImpl()
2634 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward()
2678 const DeviceMemory<float>& input_h_data, in DoRnnForward()
2721 const DeviceMemory<double>& input_h_data, in DoRnnForward()
2744 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnBackward()
2797 const DeviceMemory<float>& input_h_data, in DoRnnBackward()
2849 const DeviceMemory<double>& input_h_data, in DoRnnBackward()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1719 const DeviceMemory<T>& input_h_data, in ExtractAndCheckRnnForward()
1875 const DeviceMemory<T>& input_h_data, in DoRnnForwardImpl()
2088 const DeviceMemory<T>& input_h_data, in DoRnnBackwardImpl()
2415 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward()
2459 const DeviceMemory<float>& input_h_data, in DoRnnForward()
2502 const DeviceMemory<double>& input_h_data, in DoRnnForward()
2546 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnBackward()
2599 const DeviceMemory<float>& input_h_data, in DoRnnBackward()
2651 const DeviceMemory<double>& input_h_data, in DoRnnBackward()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc816 auto input_h_data = AsDeviceMemory<T>(input_h); in DoForward() local
899 auto input_h_data = AsDeviceMemory<T>(input_h); in DoBackward() local