/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 145 DeviceMemory<Eigen::half>* input_h_backprop_data, 170 DeviceMemory<float>* input_h_backprop_data, 195 DeviceMemory<double>* input_h_backprop_data, 643 DeviceMemory<T>* input_h_backprop_data,
|
D | cuda_dnn.cc | 1677 DeviceMemory<T>* input_h_backprop_data, in DoRnnBackwardImpl() argument 1726 /*dhx=*/input_h_backprop_data->opaque(), in DoRnnBackwardImpl() 1754 /*dhx=*/input_h_backprop_data->opaque(), in DoRnnBackwardImpl() 2015 DeviceMemory<Eigen::half>* input_h_backprop_data, in DoRnnBackward() argument 2042 input_backprop_data, input_h_backprop_data, input_c_backprop_data, in DoRnnBackward() 2066 DeviceMemory<float>* input_h_backprop_data, in DoRnnBackward() argument 2093 input_backprop_data, input_h_backprop_data, input_c_backprop_data, in DoRnnBackward() 2118 DeviceMemory<double>* input_h_backprop_data, in DoRnnBackward() argument 2145 input_backprop_data, input_h_backprop_data, input_c_backprop_data, in DoRnnBackward()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 136 DeviceMemory<Eigen::half>* input_h_backprop_data, 161 DeviceMemory<float>* input_h_backprop_data, 186 DeviceMemory<double>* input_h_backprop_data, 698 DeviceMemory<T>* input_h_backprop_data,
|
D | rocm_dnn.cc | 2134 DeviceMemory<T>* input_h_backprop_data, in DoRnnBackwardImpl() argument 2176 if ((size_data > 0) && (input_h_backprop_data->opaque() != nullptr)) in DoRnnBackwardImpl() 2177 stream->ThenMemZero(input_h_backprop_data, size_data * sizeof(float)); in DoRnnBackwardImpl() 2198 input_h_backprop_data->opaque() /*dhx*/, in DoRnnBackwardImpl() 2450 DeviceMemory<Eigen::half>* input_h_backprop_data, in DoRnnBackward() argument 2479 input_h_backprop_data, input_c_backprop_data, params_backprop_data, in DoRnnBackward() 2501 DeviceMemory<float>* input_h_backprop_data, in DoRnnBackward() argument 2530 input_h_backprop_data, input_c_backprop_data, params_backprop_data, in DoRnnBackward() 2553 DeviceMemory<double>* input_h_backprop_data, in DoRnnBackward() argument
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 2238 DeviceMemory<Eigen::half>* input_h_backprop_data, 2266 DeviceMemory<float>* input_h_backprop_data, 2294 DeviceMemory<double>* input_h_backprop_data,
|
D | stream.h | 1813 DeviceMemory<Eigen::half> *input_h_backprop_data, 1838 DeviceMemory<float> *input_h_backprop_data, 1863 DeviceMemory<double> *input_h_backprop_data,
|
D | stream.cc | 4974 DeviceMemory<Eigen::half> *input_h_backprop_data, in ThenRnnBackward() argument 4988 input_backprop_data, input_h_backprop_data, input_c_backprop_data, in ThenRnnBackward() 5020 DeviceMemory<float> *input_h_backprop_data, in ThenRnnBackward() argument 5034 input_backprop_data, input_h_backprop_data, input_c_backprop_data, in ThenRnnBackward() 5067 DeviceMemory<double> *input_h_backprop_data, in ThenRnnBackward() argument 5081 input_backprop_data, input_h_backprop_data, input_c_backprop_data, in ThenRnnBackward()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | cudnn_rnn_ops.cc | 836 auto input_h_backprop_data = AsDeviceMemory<T>(input_h_backprop); in DoBackward() local 856 &input_h_backprop_data, &input_c_backprop_data, in DoBackward()
|