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.h2189 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward()
2210 const DeviceMemory<float>& input_h_data, in DoRnnForward()
2231 const DeviceMemory<double>& input_h_data, in DoRnnForward()
2293 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnBackward()
2321 const DeviceMemory<float>& input_h_data, in DoRnnBackward()
2349 const DeviceMemory<double>& input_h_data, in DoRnnBackward()
Dstream.cc4543 const DeviceMemory<Eigen::half> &input_h_data, in ThenRnnForward()
4577 const DeviceMemory<float> &input_h_data, in ThenRnnForward()
4610 const DeviceMemory<double> &input_h_data, in ThenRnnForward()
4644 const DeviceMemory<Eigen::half> &input_h_data, in ThenRnnBackward()
4689 const DeviceMemory<float> &input_h_data, in ThenRnnBackward()
4733 const DeviceMemory<double> &input_h_data, in ThenRnnBackward()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1602 const DeviceMemory<T>& input_h_data, in ExtractAndCheckRnnForward()
1751 const DeviceMemory<T>& input_h_data, in DoRnnForwardImpl()
1891 const DeviceMemory<T>& input_h_data, in DoRnnBackwardImpl()
2131 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward()
2173 const DeviceMemory<float>& input_h_data, in DoRnnForward()
2214 const DeviceMemory<double>& input_h_data, in DoRnnForward()
2256 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnBackward()
2308 const DeviceMemory<float>& input_h_data, in DoRnnBackward()
2359 const DeviceMemory<double>& input_h_data, in DoRnnBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1977 const DeviceMemory<T>& input_h_data, in ExtractAndCheckRnnForward()
2087 const DeviceMemory<T>& input_h_data, in DoRnnForwardImpl()
2202 const DeviceMemory<T>& input_h_data, in DoRnnBackwardImpl()
2582 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward()
2625 const DeviceMemory<float>& input_h_data, in DoRnnForward()
2667 const DeviceMemory<double>& input_h_data, in DoRnnForward()
2689 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnBackward()
2741 const DeviceMemory<float>& input_h_data, in DoRnnBackward()
2792 const DeviceMemory<double>& input_h_data, in DoRnnBackward()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc814 auto input_h_data = AsDeviceMemory<T>(input_h); in DoForward() local
877 auto input_h_data = AsDeviceMemory<T>(input_h); in DoBackward() local