Searched refs:output_diff_data (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 465 DeviceMemory<double>* output_diff_data, 475 DeviceMemory<float>* output_diff_data, 485 DeviceMemory<Eigen::half>* output_diff_data,
|
D | cuda_dnn.cc | 3835 DeviceMemory<double>* output_diff_data, in DoPoolBackward() argument 3852 output_diff_data->opaque())); in DoPoolBackward() 3865 DeviceMemory<float>* output_diff_data, in DoPoolBackward() argument 3882 output_diff_data->opaque())); in DoPoolBackward() 3895 DeviceMemory<Eigen::half>* output_diff_data, in DoPoolBackward() argument 3912 output_diff_data->opaque())); in DoPoolBackward()
|
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 448 DeviceMemory<double>* output_diff_data, 458 DeviceMemory<float>* output_diff_data, 468 DeviceMemory<Eigen::half>* output_diff_data,
|
D | rocm_dnn.cc | 3551 DeviceMemory<double>* output_diff_data, in DoPoolBackward() argument 3564 DeviceMemory<float>* output_diff_data, in DoPoolBackward() argument 3643 output_diff_data->opaque(), workspace.opaque()); in DoPoolBackward() 3660 DeviceMemory<Eigen::half>* output_diff_data, in DoPoolBackward() argument 3739 output_diff_data->opaque(), workspace.opaque()); in DoPoolBackward()
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 1619 DeviceMemory<double>* output_diff_data, 1632 DeviceMemory<float>* output_diff_data, 1645 DeviceMemory<Eigen::half>* output_diff_data,
|
D | stream.cc | 1319 DeviceMemory<double> *output_diff_data, in ThenPoolBackward() argument 1323 PARAM(input_diff_data), PARAM(output_diff_data), in ThenPoolBackward() 1330 input_diff_data, output_diff_data, in ThenPoolBackward() 1349 DeviceMemory<float> *output_diff_data, in ThenPoolBackward() argument 1353 PARAM(input_diff_data), PARAM(output_diff_data), in ThenPoolBackward() 1360 input_diff_data, output_diff_data, in ThenPoolBackward() 1376 DeviceMemory<Eigen::half> *output_diff_data, in ThenPoolBackward() argument 1380 PARAM(input_diff_data), PARAM(output_diff_data), in ThenPoolBackward() 1387 input_diff_data, output_diff_data, in ThenPoolBackward()
|
D | stream.h | 551 DeviceMemory<double> *output_diff_data, 560 DeviceMemory<float> *output_diff_data, 569 DeviceMemory<Eigen::half> *output_diff_data,
|