Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h2120 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward()
2142 const DeviceMemory<float>& input_c_data, in DoRnnForward()
2164 const DeviceMemory<double>& input_c_data, in DoRnnForward()
2227 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnBackward()
2256 const DeviceMemory<float>& input_c_data, in DoRnnBackward()
2285 const DeviceMemory<double>& input_c_data, in DoRnnBackward()
H A Dstream.cc2038 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnForward()
2074 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> &params, in ThenRnnForward()
2109 const DeviceMemory<double> &input_c_data, in ThenRnnForward()
2145 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnBackward()
2191 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> &params, in ThenRnnBackward()
2236 const DeviceMemory<double> &input_c_data, in ThenRnnBackward()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc2033 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in ExtractAndCheckRnnForward()
2143 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnForwardImpl()
2258 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnBackwardImpl()
2639 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward()
2683 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnForward()
2726 const DeviceMemory<double>& input_c_data, in DoRnnForward()
2749 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnBackward()
2802 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnBackward()
2854 const DeviceMemory<double>& input_c_data, in DoRnnBackward()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc1779 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in ExtractAndCheckRnnForward()
1937 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnForwardImpl()
2150 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnBackwardImpl()
2477 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward()
2521 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnForward()
2564 const DeviceMemory<double>& input_c_data, in DoRnnForward()
2608 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnBackward()
2661 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnBackward()
2713 const DeviceMemory<double>& input_c_data, in DoRnnBackward()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcudnn_rnn_ops.cc818 DeviceMemory<T> input_c_data; in DoForward() local
901 DeviceMemory<T> input_c_data; in DoBackward() local