Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1418 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/
Drocm_dnn.cc1898 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/
Dcudnn_rnn_ops.cc756 DeviceMemory<T> input_c_data; in DoForward() local
818 DeviceMemory<T> input_c_data; in DoBackward() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc4852 const DeviceMemory<Eigen::half> &input_c_data, in ThenRnnForward()
4889 const DeviceMemory<float> &input_c_data, const DeviceMemory<float> &params, 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> &params, in ThenRnnBackward()
5055 const DeviceMemory<double> &input_c_data, in ThenRnnBackward()