Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h2118 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward()
2140 const DeviceMemory<float>& input_h_data, in DoRnnForward()
2162 const DeviceMemory<double>& input_h_data, in DoRnnForward()
2225 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnBackward()
2254 const DeviceMemory<float>& input_h_data, in DoRnnBackward()
2283 const DeviceMemory<double>& input_h_data, in DoRnnBackward()
H A Dstream.cc2036 const DeviceMemory<Eigen::half> &input_h_data, in ThenRnnForward()
2072 const DeviceMemory<float> &input_h_data, in ThenRnnForward()
2107 const DeviceMemory<double> &input_h_data, in ThenRnnForward()
2143 const DeviceMemory<Eigen::half> &input_h_data, in ThenRnnBackward()
2189 const DeviceMemory<float> &input_h_data, in ThenRnnBackward()
2234 const DeviceMemory<double> &input_h_data, in ThenRnnBackward()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc2031 const DeviceMemory<T>& input_h_data, in ExtractAndCheckRnnForward()
2141 const DeviceMemory<T>& input_h_data, in DoRnnForwardImpl()
2256 const DeviceMemory<T>& input_h_data, in DoRnnBackwardImpl()
2637 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward()
2681 const DeviceMemory<float>& input_h_data, in DoRnnForward()
2724 const DeviceMemory<double>& input_h_data, in DoRnnForward()
2747 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnBackward()
2800 const DeviceMemory<float>& input_h_data, in DoRnnBackward()
2852 const DeviceMemory<double>& input_h_data, in DoRnnBackward()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc1777 const DeviceMemory<T>& input_h_data, in ExtractAndCheckRnnForward()
1935 const DeviceMemory<T>& input_h_data, in DoRnnForwardImpl()
2148 const DeviceMemory<T>& input_h_data, in DoRnnBackwardImpl()
2475 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnForward()
2519 const DeviceMemory<float>& input_h_data, in DoRnnForward()
2562 const DeviceMemory<double>& input_h_data, in DoRnnForward()
2606 const DeviceMemory<Eigen::half>& input_h_data, in DoRnnBackward()
2659 const DeviceMemory<float>& input_h_data, in DoRnnBackward()
2711 const DeviceMemory<double>& input_h_data, in DoRnnBackward()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcudnn_rnn_ops.cc817 auto input_h_data = AsDeviceMemory<T>(input_h); in DoForward() local
900 auto input_h_data = AsDeviceMemory<T>(input_h); in DoBackward() local