Searched defs:input_c_data (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1942 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in ExtractAndCheckRnnForward()
2052 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnForwardImpl()
2167 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnBackwardImpl()
2377 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward()
2420 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnForward()
2462 const DeviceMemory<double>& input_c_data, in DoRnnForward()
2484 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnBackward()
2536 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnBackward()
2587 const DeviceMemory<double>& input_c_data, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1579 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in ExtractAndCheckRnnForward()
1728 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnForwardImpl()
1880 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnBackwardImpl()
2130 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward()
2172 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnForward()
2213 const DeviceMemory<double>& input_c_data, in DoRnnForward()
2255 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnBackward()
2307 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnBackward()
2358 const DeviceMemory<double>& input_c_data, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 811 DeviceMemory<T> input_c_data; in DoForward() local
874 DeviceMemory<T> input_c_data; in DoBackward() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 4994 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnForward()
5031 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> ¶ms, in ThenRnnForward()
5067 const DeviceMemory<double> &input_c_data, in ThenRnnForward()
5104 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnBackward()
5151 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> ¶ms, in ThenRnnBackward()
5197 const DeviceMemory<double> &input_c_data, in ThenRnnBackward()
|