Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.h158 DeviceMemory<Eigen::half>* input_c_backprop_data,
184 DeviceMemory<float>* input_c_backprop_data,
210 DeviceMemory<double>* input_c_backprop_data,
607 DeviceMemory<T>* input_c_backprop_data,
H A Dcuda_dnn.cc2162 DeviceMemory<T>* input_c_backprop_data, in DoRnnBackwardImpl() argument
2221 /*dcx=*/input_c_backprop_data->opaque(), in DoRnnBackwardImpl()
2297 /*dcx=*/input_c_backprop_data->opaque(), nullptr, nullptr, in DoRnnBackwardImpl()
2320 /*dcx=*/input_c_backprop_data->opaque(), in DoRnnBackwardImpl()
2621 DeviceMemory<Eigen::half>* input_c_backprop_data, in DoRnnBackward() argument
2648 input_c_backprop_data, params_backprop_data, reserve_space_data, in DoRnnBackward()
2673 DeviceMemory<float>* input_c_backprop_data, in DoRnnBackward() argument
2700 input_c_backprop_data, params_backprop_data, reserve_space_data, in DoRnnBackward()
2726 DeviceMemory<double>* input_c_backprop_data, in DoRnnBackward() argument
2753 input_c_backprop_data, params_backprop_data, reserve_space_data, in DoRnnBackward()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.h178 DeviceMemory<Eigen::half>* input_c_backprop_data,
204 DeviceMemory<float>* input_c_backprop_data,
230 DeviceMemory<double>* input_c_backprop_data,
702 DeviceMemory<T>* input_c_backprop_data,
H A Drocm_dnn.cc2270 DeviceMemory<T>* input_c_backprop_data, in DoRnnBackwardImpl() argument
2317 if ((size_data > 0) && (input_c_backprop_data->opaque() != nullptr)) in DoRnnBackwardImpl()
2318 stream->ThenMemZero(input_c_backprop_data, size_data * type_size); in DoRnnBackwardImpl()
2336 input_c_backprop_data->opaque() /*dcx*/, workspace.opaque() /*workspace*/, in DoRnnBackwardImpl()
2762 DeviceMemory<Eigen::half>* input_c_backprop_data, in DoRnnBackward() argument
2790 input_h_backprop_data, input_c_backprop_data, params_backprop_data, in DoRnnBackward()
2814 DeviceMemory<float>* input_c_backprop_data, in DoRnnBackward() argument
2842 input_h_backprop_data, input_c_backprop_data, params_backprop_data, in DoRnnBackward()
2867 DeviceMemory<double>* input_c_backprop_data, in DoRnnBackward() argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream.cc2158 DeviceMemory<Eigen::half> *input_c_backprop_data, in ThenRnnBackward() argument
2170 input_backprop_data, input_h_backprop_data, input_c_backprop_data, in ThenRnnBackward()
2203 DeviceMemory<float> *input_c_backprop_data, in ThenRnnBackward() argument
2215 input_backprop_data, input_h_backprop_data, input_c_backprop_data, in ThenRnnBackward()
2249 DeviceMemory<double> *input_c_backprop_data, in ThenRnnBackward() argument
2261 input_backprop_data, input_h_backprop_data, input_c_backprop_data, in ThenRnnBackward()
H A Ddnn.h2240 DeviceMemory<Eigen::half>* input_c_backprop_data, in DoRnnBackward() argument
2269 DeviceMemory<float>* input_c_backprop_data, in DoRnnBackward() argument
2298 DeviceMemory<double>* input_c_backprop_data, in DoRnnBackward() argument
H A Dstream.h1307 DeviceMemory<Eigen::half> *input_c_backprop_data,
1333 DeviceMemory<float> *input_c_backprop_data,
1359 DeviceMemory<double> *input_c_backprop_data,
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcudnn_rnn_ops.cc920 DeviceMemory<T> input_c_backprop_data; in DoBackward() local
922 input_c_backprop_data = AsDeviceMemory<T>(input_c_backprop); in DoBackward()
959 &input_c_backprop_data, &params_backprop_data, in DoBackward()