Searched defs:input_c_data (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1418 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in ExtractAndCheckRnnForward()
1513 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnForwardImpl()
1666 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnBackwardImpl()
1878 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward()
1920 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnForward()
1961 const DeviceMemory<double>& input_c_data, in DoRnnForward()
2003 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnBackward()
2055 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnBackward()
2106 const DeviceMemory<double>& input_c_data, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1898 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in ExtractAndCheckRnnForward()
2008 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnForwardImpl()
2123 const DeviceMemory<T>& input_c_data, const DeviceMemory<T>& params, in DoRnnBackwardImpl()
2331 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnForward()
2374 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnForward()
2416 const DeviceMemory<double>& input_c_data, in DoRnnForward()
2438 const DeviceMemory<Eigen::half>& input_c_data, in DoRnnBackward()
2490 const DeviceMemory<float>& input_c_data, const DeviceMemory<float>& params, in DoRnnBackward()
2541 const DeviceMemory<double>& input_c_data, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 756 DeviceMemory<T> input_c_data; in DoForward() local
818 DeviceMemory<T> input_c_data; in DoBackward() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 4852 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnForward()
4889 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> ¶ms, in ThenRnnForward()
4925 const DeviceMemory<double> &input_c_data, in ThenRnnForward()
4962 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnBackward()
5009 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> ¶ms, in ThenRnnBackward()
5055 const DeviceMemory<double> &input_c_data, in ThenRnnBackward()
|