Searched refs:input_diff_data (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/rocm/ |
D | rocm_dnn.h | 499 const DeviceMemory<double>& input_diff_data, 509 const DeviceMemory<float>& input_diff_data, 519 const DeviceMemory<Eigen::half>& input_diff_data, 893 const DeviceMemory<T>& input_diff_data,
|
D | rocm_dnn.cc | 4197 const DeviceMemory<T>& output_data, const DeviceMemory<T>& input_diff_data, in DoPoolBackwardImpl() argument 4290 output_data.opaque(), dest_desc.handle(), input_diff_data.opaque(), in DoPoolBackwardImpl() 4309 const DeviceMemory<double>& input_diff_data, in DoPoolBackward() argument 4322 const DeviceMemory<float>& input_diff_data, in DoPoolBackward() argument 4327 input_diff_data, output_diff_data, in DoPoolBackward() 4337 const DeviceMemory<Eigen::half>& input_diff_data, in DoPoolBackward() argument 4342 input_diff_data, output_diff_data, in DoPoolBackward()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | cuda_dnn.h | 483 const DeviceMemory<double>& input_diff_data, 493 const DeviceMemory<float>& input_diff_data, 503 const DeviceMemory<Eigen::half>& input_diff_data,
|
D | cuda_dnn.cc | 4367 const DeviceMemory<double>& input_diff_data, in DoPoolBackward() argument 4383 output_data.opaque(), dest_desc.handle(), input_diff_data.opaque(), in DoPoolBackward() 4397 const DeviceMemory<float>& input_diff_data, in DoPoolBackward() argument 4413 output_data.opaque(), dest_desc.handle(), input_diff_data.opaque(), in DoPoolBackward() 4427 const DeviceMemory<Eigen::half>& input_diff_data, in DoPoolBackward() argument 4443 output_data.opaque(), dest_desc.handle(), input_diff_data.opaque(), in DoPoolBackward()
|
/external/tensorflow/tensorflow/stream_executor/ |
D | dnn.h | 1684 const DeviceMemory<double>& input_diff_data, in DoPoolBackward() argument 1697 const DeviceMemory<float>& input_diff_data, in DoPoolBackward() argument 1710 const DeviceMemory<Eigen::half>& input_diff_data, in DoPoolBackward() argument
|
D | stream.cc | 908 const DeviceMemory<double> &input_diff_data, in ThenPoolBackward() argument 913 PARAM(input_diff_data), PARAM(output_diff_data), in ThenPoolBackward() 919 input_diff_data, output_diff_data, in ThenPoolBackward() 935 const DeviceMemory<float> &input_diff_data, in ThenPoolBackward() argument 940 PARAM(input_diff_data), PARAM(output_diff_data), in ThenPoolBackward() 946 input_diff_data, output_diff_data, in ThenPoolBackward() 960 const DeviceMemory<Eigen::half> &input_diff_data, in ThenPoolBackward() argument 965 PARAM(input_diff_data), PARAM(output_diff_data), in ThenPoolBackward() 971 input_diff_data, output_diff_data, in ThenPoolBackward()
|
D | stream.h | 573 const DeviceMemory<double> &input_diff_data, 582 const DeviceMemory<float> &input_diff_data, 591 const DeviceMemory<Eigen::half> &input_diff_data,
|