Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3840 DeviceMemory<double>* output_diff_data, in DoPoolBackward()
3853 DeviceMemory<float>* output_diff_data, in DoPoolBackward()
3949 DeviceMemory<Eigen::half>* output_diff_data, in DoPoolBackward()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc4377 DeviceMemory<double>* output_diff_data, in DoPoolBackward()
4407 DeviceMemory<float>* output_diff_data, in DoPoolBackward()
4437 DeviceMemory<Eigen::half>* output_diff_data, in DoPoolBackward()
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc1461 DeviceMemory<double> *output_diff_data, in ThenPoolBackward()
1491 DeviceMemory<float> *output_diff_data, in ThenPoolBackward()
1518 DeviceMemory<Eigen::half> *output_diff_data, in ThenPoolBackward()