/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 86 DeviceMemory<Eigen::half>* output_c_data, bool is_training, 104 DeviceMemory<float>* output_c_data, bool is_training, 122 DeviceMemory<double>* output_c_data, bool is_training, 140 const DeviceMemory<Eigen::half>& output_c_data, 165 const DeviceMemory<float>& output_c_data, 190 const DeviceMemory<double>& output_c_data, 619 DeviceMemory<T>* output_c_data, bool is_training, 638 const DeviceMemory<T>& output_c_data,
|
D | cuda_dnn.cc | 1424 const DeviceMemory<T>& output_c_data) { in ExtractAndCheckRnnForward() argument 1519 DeviceMemory<T>* output_c_data, bool is_training, in DoRnnForwardImpl() argument 1528 output_h_desc, *output_h_data, output_c_desc, *output_c_data)); in DoRnnForwardImpl() 1578 /*cyDesc=*/output_c_desc.handle(), /*cy=*/output_c_data->opaque(), in DoRnnForwardImpl() 1599 /*cy=*/output_c_data->opaque(), /*workspace=*/workspace.opaque(), in DoRnnForwardImpl() 1615 /*cyDesc=*/output_c_desc.handle(), /*cy=*/output_c_data->opaque(), in DoRnnForwardImpl() 1638 /*cy=*/output_c_data->opaque(), /*workspace=*/workspace.opaque(), in DoRnnForwardImpl() 1672 const DeviceMemory<T>& output_c_data, in DoRnnBackwardImpl() argument 1688 output_h_data, output_c_desc, output_c_data)); in DoRnnBackwardImpl() 1885 DeviceMemory<Eigen::half>* output_c_data, bool is_training, in DoRnnForward() argument [all …]
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 77 DeviceMemory<Eigen::half>* output_c_data, bool is_training, 95 DeviceMemory<float>* output_c_data, bool is_training, 113 DeviceMemory<double>* output_c_data, bool is_training, 131 const DeviceMemory<Eigen::half>& output_c_data, 156 const DeviceMemory<float>& output_c_data, 181 const DeviceMemory<double>& output_c_data, 676 DeviceMemory<T>* output_c_data, bool is_training, 693 const DeviceMemory<T>& output_c_data,
|
D | rocm_dnn.cc | 1904 const DeviceMemory<T>& output_c_data, RnnModelDims* model_dims) { in ExtractAndCheckRnnForward() argument 2014 DeviceMemory<T>* output_c_data, bool is_training, in DoRnnForwardImpl() argument 2022 output_h_desc, *output_h_data, output_c_desc, *output_c_data, in DoRnnForwardImpl() 2084 output_c_data->opaque() /*cy*/, workspace.opaque() /*workspace*/, in DoRnnForwardImpl() 2102 output_c_data->opaque() /*cy*/, workspace.opaque() /*workspace*/, in DoRnnForwardImpl() 2129 const DeviceMemory<T>& output_c_data, in DoRnnBackwardImpl() argument 2144 output_h_desc, output_h_data, output_c_desc, output_c_data, &model_dims); in DoRnnBackwardImpl() 2338 DeviceMemory<Eigen::half>* output_c_data, bool is_training, in DoRnnForward() argument 2363 output_h_data, miopen_output_c_desc, output_c_data, is_training, in DoRnnForward() 2380 DeviceMemory<float>* output_c_data, bool is_training, in DoRnnForward() argument [all …]
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 2129 DeviceMemory<Eigen::half>* output_c_data, 2150 DeviceMemory<float>* output_c_data, 2171 DeviceMemory<double>* output_c_data, 2233 const DeviceMemory<Eigen::half>& output_c_data, 2261 const DeviceMemory<float>& output_c_data, 2289 const DeviceMemory<double>& output_c_data,
|
D | stream.h | 1750 DeviceMemory<Eigen::half> *output_c_data, 1769 DeviceMemory<float> *output_c_data, bool is_training, 1787 DeviceMemory<double> *output_c_data, bool is_training, 1808 const DeviceMemory<Eigen::half> &output_c_data, 1833 const DeviceMemory<float> &output_c_data, 1858 const DeviceMemory<double> &output_c_data,
|
D | stream.cc | 4859 DeviceMemory<Eigen::half> *output_c_data, bool is_training, in ThenRnnForward() argument 4869 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward() 4895 DeviceMemory<float> *output_c_data, bool is_training, in ThenRnnForward() argument 4905 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward() 4932 DeviceMemory<double> *output_c_data, bool is_training, in ThenRnnForward() argument 4942 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnForward() 4969 const DeviceMemory<Eigen::half> &output_c_data, in ThenRnnBackward() argument 4986 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward() 5015 const DeviceMemory<float> &output_c_data, in ThenRnnBackward() argument 5032 output_h_desc, output_h_data, output_c_desc, output_c_data, in ThenRnnBackward() [all …]
|
/external/tensorflow/tensorflow/core/kernels/ |
D | cudnn_rnn_ops.cc | 764 DeviceMemory<T> output_c_data; in DoForward() local 766 output_c_data = AsDeviceMemory<T>(output_c); in DoForward() 775 &output_h_data, *state_desc, &output_c_data, in DoForward() 825 DeviceMemory<T> output_c_data; in DoBackward() local 827 output_c_data = AsDeviceMemory<T>(output_c); in DoBackward() 853 output_h_data, *state_desc, output_c_data, in DoBackward()
|