Searched defs:input_diff_data (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 1585 const DeviceMemory<double>& input_diff_data, in DoPoolBackward()
1598 const DeviceMemory<float>& input_diff_data, in DoPoolBackward()
1611 const DeviceMemory<Eigen::half>& input_diff_data, in DoPoolBackward()
|
| D | stream.cc | 705 const DeviceMemory<double> &input_diff_data, in ThenPoolBackward()
732 const DeviceMemory<float> &input_diff_data, in ThenPoolBackward()
757 const DeviceMemory<Eigen::half> &input_diff_data, in ThenPoolBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 4088 const DeviceMemory<T>& output_data, const DeviceMemory<T>& input_diff_data, in DoPoolBackwardImpl()
4200 const DeviceMemory<double>& input_diff_data, in DoPoolBackward()
4213 const DeviceMemory<float>& input_diff_data, in DoPoolBackward()
4228 const DeviceMemory<Eigen::half>& input_diff_data, in DoPoolBackward()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 5575 const DeviceMemory<double>& input_diff_data, in DoPoolBackward()
5605 const DeviceMemory<float>& input_diff_data, in DoPoolBackward()
5635 const DeviceMemory<Eigen::half>& input_diff_data, in DoPoolBackward()
|