Home
last modified time | relevance | path

Searched refs:input_diff_data (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h499 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,
Drocm_dnn.cc4197 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/
Dcuda_dnn.h483 const DeviceMemory<double>& input_diff_data,
493 const DeviceMemory<float>& input_diff_data,
503 const DeviceMemory<Eigen::half>& input_diff_data,
Dcuda_dnn.cc4367 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/
Ddnn.h1684 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
Dstream.cc908 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()
Dstream.h573 const DeviceMemory<double> &input_diff_data,
582 const DeviceMemory<float> &input_diff_data,
591 const DeviceMemory<Eigen::half> &input_diff_data,