Searched defs:output_c_data (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.cc | 1424 const DeviceMemory<T>& output_c_data) { in ExtractAndCheckRnnForward() 1519 DeviceMemory<T>* output_c_data, bool is_training, in DoRnnForwardImpl() 1672 const DeviceMemory<T>& output_c_data, in DoRnnBackwardImpl() 1885 DeviceMemory<Eigen::half>* output_c_data, bool is_training, in DoRnnForward() 1926 DeviceMemory<float>* output_c_data, bool is_training, in DoRnnForward() 1968 DeviceMemory<double>* output_c_data, bool is_training, in DoRnnForward() 2010 const DeviceMemory<Eigen::half>& output_c_data, in DoRnnBackward() 2061 const DeviceMemory<float>& output_c_data, in DoRnnBackward() 2113 const DeviceMemory<double>& output_c_data, in DoRnnBackward()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.cc | 1904 const DeviceMemory<T>& output_c_data, RnnModelDims* model_dims) { in ExtractAndCheckRnnForward() 2014 DeviceMemory<T>* output_c_data, bool is_training, in DoRnnForwardImpl() 2129 const DeviceMemory<T>& output_c_data, in DoRnnBackwardImpl() 2338 DeviceMemory<Eigen::half>* output_c_data, bool is_training, in DoRnnForward() 2380 DeviceMemory<float>* output_c_data, bool is_training, in DoRnnForward() 2423 DeviceMemory<double>* output_c_data, bool is_training, in DoRnnForward() 2445 const DeviceMemory<Eigen::half>& output_c_data, in DoRnnBackward() 2496 const DeviceMemory<float>& output_c_data, in DoRnnBackward() 2548 const DeviceMemory<double>& output_c_data, in DoRnnBackward()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | cudnn_rnn_ops.cc | 764 DeviceMemory<T> output_c_data; in DoForward() local 825 DeviceMemory<T> output_c_data; in DoBackward() local
|
/external/tensorflow/tensorflow/stream_executor/ |
D | stream.cc | 4859 DeviceMemory<Eigen::half> *output_c_data, bool is_training, in ThenRnnForward() 4895 DeviceMemory<float> *output_c_data, bool is_training, in ThenRnnForward() 4932 DeviceMemory<double> *output_c_data, bool is_training, in ThenRnnForward() 4969 const DeviceMemory<Eigen::half> &output_c_data, in ThenRnnBackward() 5015 const DeviceMemory<float> &output_c_data, in ThenRnnBackward() 5062 const DeviceMemory<double> &output_c_data, in ThenRnnBackward()
|