Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc3834 const DeviceMemory<double>& input_diff_data, in DoPoolBackward()
3864 const DeviceMemory<float>& input_diff_data, in DoPoolBackward()
3894 const DeviceMemory<Eigen::half>& input_diff_data, in DoPoolBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3550 const DeviceMemory<double>& input_diff_data, in DoPoolBackward()
3563 const DeviceMemory<float>& input_diff_data, in DoPoolBackward()
3659 const DeviceMemory<Eigen::half>& input_diff_data, in DoPoolBackward()
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc1318 const DeviceMemory<double> &input_diff_data, in ThenPoolBackward()
1348 const DeviceMemory<float> &input_diff_data, in ThenPoolBackward()
1375 const DeviceMemory<Eigen::half> &input_diff_data, in ThenPoolBackward()