Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2196 DeviceMemory<Eigen::half>* output_h_data, in DoRnnForward()
2217 DeviceMemory<float>* output_h_data, in DoRnnForward()
2238 DeviceMemory<double>* output_h_data, in DoRnnForward()
2300 const DeviceMemory<Eigen::half>& output_h_data, in DoRnnBackward()
2328 const DeviceMemory<float>& output_h_data, in DoRnnBackward()
2356 const DeviceMemory<double>& output_h_data, in DoRnnBackward()
Dstream.cc4550 DeviceMemory<Eigen::half> *output_h_data, in ThenRnnForward()
4583 DeviceMemory<float> *output_h_data, in ThenRnnForward()
4617 DeviceMemory<double> *output_h_data, in ThenRnnForward()
4651 const DeviceMemory<Eigen::half> &output_h_data, in ThenRnnBackward()
4695 const DeviceMemory<float> &output_h_data, in ThenRnnBackward()
4740 const DeviceMemory<double> &output_h_data, in ThenRnnBackward()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1608 const DeviceMemory<T>& output_h_data, in ExtractAndCheckRnnForward()
1757 DeviceMemory<T>* output_h_data, in DoRnnForwardImpl()
1897 const DeviceMemory<T>& output_h_data, in DoRnnBackwardImpl()
2138 DeviceMemory<Eigen::half>* output_h_data, in DoRnnForward()
2179 DeviceMemory<float>* output_h_data, in DoRnnForward()
2221 DeviceMemory<double>* output_h_data, in DoRnnForward()
2263 const DeviceMemory<Eigen::half>& output_h_data, in DoRnnBackward()
2314 const DeviceMemory<float>& output_h_data, in DoRnnBackward()
2366 const DeviceMemory<double>& output_h_data, in DoRnnBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1983 const DeviceMemory<T>& output_h_data, in ExtractAndCheckRnnForward()
2093 DeviceMemory<T>* output_h_data, in DoRnnForwardImpl()
2208 const DeviceMemory<T>& output_h_data, in DoRnnBackwardImpl()
2589 DeviceMemory<Eigen::half>* output_h_data, in DoRnnForward()
2631 DeviceMemory<float>* output_h_data, in DoRnnForward()
2674 DeviceMemory<double>* output_h_data, in DoRnnForward()
2696 const DeviceMemory<Eigen::half>& output_h_data, in DoRnnBackward()
2747 const DeviceMemory<float>& output_h_data, in DoRnnBackward()
2799 const DeviceMemory<double>& output_h_data, in DoRnnBackward()
/external/tensorflow/tensorflow/core/kernels/
Dcudnn_rnn_ops.cc822 auto output_h_data = AsDeviceMemory<T>(output_h); in DoForward() local
884 auto output_h_data = AsDeviceMemory<T>(output_h); in DoBackward() local