Home
last modified time | relevance | path

Searched refs:input_h_backprop_data (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h145 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,
Dcuda_dnn.cc1677 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/
Drocm_dnn.h136 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,
Drocm_dnn.cc2134 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/
Ddnn.h2238 DeviceMemory<Eigen::half>* input_h_backprop_data,
2266 DeviceMemory<float>* input_h_backprop_data,
2294 DeviceMemory<double>* input_h_backprop_data,
Dstream.h1813 DeviceMemory<Eigen::half> *input_h_backprop_data,
1838 DeviceMemory<float> *input_h_backprop_data,
1863 DeviceMemory<double> *input_h_backprop_data,
Dstream.cc4974 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/
Dcudnn_rnn_ops.cc836 auto input_h_backprop_data = AsDeviceMemory<T>(input_h_backprop); in DoBackward() local
856 &input_h_backprop_data, &input_c_backprop_data, in DoBackward()