Searched defs:output_c_data (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1948 const DeviceMemory<T>& output_c_data, RnnModelDims* model_dims) { in ExtractAndCheckRnnForward() 2058 DeviceMemory<T>* output_c_data, bool is_training, in DoRnnForwardImpl() 2173 const DeviceMemory<T>& output_c_data, in DoRnnBackwardImpl() 2384 DeviceMemory<Eigen::half>* output_c_data, bool is_training, in DoRnnForward() 2426 DeviceMemory<float>* output_c_data, bool is_training, in DoRnnForward() 2469 DeviceMemory<double>* output_c_data, bool is_training, in DoRnnForward() 2491 const DeviceMemory<Eigen::half>& output_c_data, in DoRnnBackward() 2542 const DeviceMemory<float>& output_c_data, in DoRnnBackward() 2594 const DeviceMemory<double>& output_c_data, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1585 const DeviceMemory<T>& output_c_data) { in ExtractAndCheckRnnForward() 1734 DeviceMemory<T>* output_c_data, bool is_training, in DoRnnForwardImpl() 1886 const DeviceMemory<T>& output_c_data, in DoRnnBackwardImpl() 2137 DeviceMemory<Eigen::half>* output_c_data, bool is_training, in DoRnnForward() 2178 DeviceMemory<float>* output_c_data, bool is_training, in DoRnnForward() 2220 DeviceMemory<double>* output_c_data, bool is_training, in DoRnnForward() 2262 const DeviceMemory<Eigen::half>& output_c_data, in DoRnnBackward() 2313 const DeviceMemory<float>& output_c_data, in DoRnnBackward() 2365 const DeviceMemory<double>& output_c_data, in DoRnnBackward()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_rnn_ops.cc | 819 DeviceMemory<T> output_c_data; in DoForward() local 881 DeviceMemory<T> output_c_data; in DoBackward() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 5001 DeviceMemory<Eigen::half> *output_c_data, bool is_training, in ThenRnnForward() 5037 DeviceMemory<float> *output_c_data, bool is_training, in ThenRnnForward() 5074 DeviceMemory<double> *output_c_data, bool is_training, in ThenRnnForward() 5111 const DeviceMemory<Eigen::half> &output_c_data, in ThenRnnBackward() 5157 const DeviceMemory<float> &output_c_data, in ThenRnnBackward() 5204 const DeviceMemory<double> &output_c_data, in ThenRnnBackward()
|