Searched defs:input_h_data (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 2189 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()
|
D | stream.cc | 4543 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/ |
D | cuda_dnn.cc | 1602 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/ |
D | rocm_dnn.cc | 1977 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/ |
D | cudnn_rnn_ops.cc | 814 auto input_h_data = AsDeviceMemory<T>(input_h); in DoForward() local 877 auto input_h_data = AsDeviceMemory<T>(input_h); in DoBackward() local
|