Home
last modified time | relevance | path

Searched defs:output_diff_data (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3835 DeviceMemory<double>* output_diff_data, in DoPoolBackward()
3865 DeviceMemory<float>* output_diff_data, in DoPoolBackward()
3895 DeviceMemory<Eigen::half>* output_diff_data, in DoPoolBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3551 DeviceMemory<double>* output_diff_data, in DoPoolBackward()
3564 DeviceMemory<float>* output_diff_data, in DoPoolBackward()
3660 DeviceMemory<Eigen::half>* output_diff_data, in DoPoolBackward()
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc1319 DeviceMemory<double> *output_diff_data, in ThenPoolBackward()
1349 DeviceMemory<float> *output_diff_data, in ThenPoolBackward()
1376 DeviceMemory<Eigen::half> *output_diff_data, in ThenPoolBackward()