Home
last modified time | relevance | path

Searched refs:output_diff_data (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h465 DeviceMemory<double>* output_diff_data,
475 DeviceMemory<float>* output_diff_data,
485 DeviceMemory<Eigen::half>* output_diff_data,
Dcuda_dnn.cc3835 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/
Drocm_dnn.h448 DeviceMemory<double>* output_diff_data,
458 DeviceMemory<float>* output_diff_data,
468 DeviceMemory<Eigen::half>* output_diff_data,
Drocm_dnn.cc3551 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/
Ddnn.h1619 DeviceMemory<double>* output_diff_data,
1632 DeviceMemory<float>* output_diff_data,
1645 DeviceMemory<Eigen::half>* output_diff_data,
Dstream.cc1319 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()
Dstream.h551 DeviceMemory<double> *output_diff_data,
560 DeviceMemory<float> *output_diff_data,
569 DeviceMemory<Eigen::half> *output_diff_data,