Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h2125 DeviceMemory<Eigen::half>* output_h_data, in DoRnnForward()
2147 DeviceMemory<float>* output_h_data, in DoRnnForward()
2169 DeviceMemory<double>* output_h_data, in DoRnnForward()
2232 const DeviceMemory<Eigen::half>& output_h_data, in DoRnnBackward()
2261 const DeviceMemory<float>& output_h_data, in DoRnnBackward()
2290 const DeviceMemory<double>& output_h_data, in DoRnnBackward()
H A Dstream.cc2043 DeviceMemory<Eigen::half> *output_h_data, in ThenRnnForward()
2078 DeviceMemory<float> *output_h_data, in ThenRnnForward()
2114 DeviceMemory<double> *output_h_data, in ThenRnnForward()
2150 const DeviceMemory<Eigen::half> &output_h_data, in ThenRnnBackward()
2195 const DeviceMemory<float> &output_h_data, in ThenRnnBackward()
2241 const DeviceMemory<double> &output_h_data, in ThenRnnBackward()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc2037 const DeviceMemory<T>& output_h_data, in ExtractAndCheckRnnForward()
2147 DeviceMemory<T>* output_h_data, in DoRnnForwardImpl()
2262 const DeviceMemory<T>& output_h_data, in DoRnnBackwardImpl()
2644 DeviceMemory<Eigen::half>* output_h_data, in DoRnnForward()
2687 DeviceMemory<float>* output_h_data, in DoRnnForward()
2731 DeviceMemory<double>* output_h_data, in DoRnnForward()
2754 const DeviceMemory<Eigen::half>& output_h_data, in DoRnnBackward()
2806 const DeviceMemory<float>& output_h_data, in DoRnnBackward()
2859 const DeviceMemory<double>& output_h_data, in DoRnnBackward()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc1783 const DeviceMemory<T>& output_h_data, in ExtractAndCheckRnnForward()
1941 DeviceMemory<T>* output_h_data, in DoRnnForwardImpl()
2154 const DeviceMemory<T>& output_h_data, in DoRnnBackwardImpl()
2482 DeviceMemory<Eigen::half>* output_h_data, in DoRnnForward()
2525 DeviceMemory<float>* output_h_data, in DoRnnForward()
2569 DeviceMemory<double>* output_h_data, in DoRnnForward()
2613 const DeviceMemory<Eigen::half>& output_h_data, in DoRnnBackward()
2665 const DeviceMemory<float>& output_h_data, in DoRnnBackward()
2718 const DeviceMemory<double>& output_h_data, in DoRnnBackward()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcudnn_rnn_ops.cc825 auto output_h_data = AsDeviceMemory<T>(output_h); in DoForward() local
907 auto output_h_data = AsDeviceMemory<T>(output_h); in DoBackward() local