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()
|