Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h146 DeviceMemory<Eigen::half>* input_c_backprop_data,
171 DeviceMemory<float>* input_c_backprop_data,
196 DeviceMemory<double>* input_c_backprop_data,
644 DeviceMemory<T>* input_c_backprop_data,
Dcuda_dnn.cc1678 DeviceMemory<T>* input_c_backprop_data, in DoRnnBackwardImpl() argument
1728 /*dcx=*/input_c_backprop_data->opaque(), nullptr, nullptr, in DoRnnBackwardImpl()
1756 /*dcx=*/input_c_backprop_data->opaque(), in DoRnnBackwardImpl()
2016 DeviceMemory<Eigen::half>* input_c_backprop_data, in DoRnnBackward() argument
2042 input_backprop_data, input_h_backprop_data, input_c_backprop_data, in DoRnnBackward()
2067 DeviceMemory<float>* input_c_backprop_data, in DoRnnBackward() argument
2093 input_backprop_data, input_h_backprop_data, input_c_backprop_data, in DoRnnBackward()
2119 DeviceMemory<double>* input_c_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.h137 DeviceMemory<Eigen::half>* input_c_backprop_data,
162 DeviceMemory<float>* input_c_backprop_data,
187 DeviceMemory<double>* input_c_backprop_data,
699 DeviceMemory<T>* input_c_backprop_data,
Drocm_dnn.cc2135 DeviceMemory<T>* input_c_backprop_data, in DoRnnBackwardImpl() argument
2181 if ((size_data > 0) && (input_c_backprop_data->opaque() != nullptr)) in DoRnnBackwardImpl()
2182 stream->ThenMemZero(input_c_backprop_data, size_data * sizeof(float)); in DoRnnBackwardImpl()
2200 input_c_backprop_data->opaque() /*dcx*/, workspace.opaque() /*workspace*/, in DoRnnBackwardImpl()
2451 DeviceMemory<Eigen::half>* input_c_backprop_data, in DoRnnBackward() argument
2479 input_h_backprop_data, input_c_backprop_data, params_backprop_data, in DoRnnBackward()
2502 DeviceMemory<float>* input_c_backprop_data, in DoRnnBackward() argument
2530 input_h_backprop_data, input_c_backprop_data, params_backprop_data, in DoRnnBackward()
2554 DeviceMemory<double>* input_c_backprop_data, in DoRnnBackward() argument
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h2239 DeviceMemory<Eigen::half>* input_c_backprop_data,
2267 DeviceMemory<float>* input_c_backprop_data,
2295 DeviceMemory<double>* input_c_backprop_data,
Dstream.h1814 DeviceMemory<Eigen::half> *input_c_backprop_data,
1839 DeviceMemory<float> *input_c_backprop_data,
1864 DeviceMemory<double> *input_c_backprop_data,
Dstream.cc4975 DeviceMemory<Eigen::half> *input_c_backprop_data, in ThenRnnBackward() argument
4988 input_backprop_data, input_h_backprop_data, input_c_backprop_data, in ThenRnnBackward()
5021 DeviceMemory<float> *input_c_backprop_data, in ThenRnnBackward() argument
5034 input_backprop_data, input_h_backprop_data, input_c_backprop_data, in ThenRnnBackward()
5068 DeviceMemory<double> *input_c_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.cc837 DeviceMemory<T> input_c_backprop_data; in DoBackward() local
839 input_c_backprop_data = AsDeviceMemory<T>(input_c_backprop); in DoBackward()
856 &input_h_backprop_data, &input_c_backprop_data, in DoBackward()