Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h464 const DeviceMemory<double>& input_diff_data,
474 const DeviceMemory<float>& input_diff_data,
484 const DeviceMemory<Eigen::half>& input_diff_data,
Dcuda_dnn.cc3834 const DeviceMemory<double>& input_diff_data, in DoPoolBackward() argument
3850 output_data.opaque(), dest_desc.handle(), input_diff_data.opaque(), in DoPoolBackward()
3864 const DeviceMemory<float>& input_diff_data, in DoPoolBackward() argument
3880 output_data.opaque(), dest_desc.handle(), input_diff_data.opaque(), in DoPoolBackward()
3894 const DeviceMemory<Eigen::half>& input_diff_data, in DoPoolBackward() argument
3910 output_data.opaque(), dest_desc.handle(), input_diff_data.opaque(), in DoPoolBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h447 const DeviceMemory<double>& input_diff_data,
457 const DeviceMemory<float>& input_diff_data,
467 const DeviceMemory<Eigen::half>& input_diff_data,
Drocm_dnn.cc3550 const DeviceMemory<double>& input_diff_data, in DoPoolBackward() argument
3563 const DeviceMemory<float>& input_diff_data, in DoPoolBackward() argument
3641 dest2.opaque(), dest_desc.handle(), input_diff_data.opaque(), in DoPoolBackward()
3659 const DeviceMemory<Eigen::half>& input_diff_data, in DoPoolBackward() argument
3737 dest2.opaque(), dest_desc.handle(), input_diff_data.opaque(), in DoPoolBackward()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1618 const DeviceMemory<double>& input_diff_data,
1631 const DeviceMemory<float>& input_diff_data,
1644 const DeviceMemory<Eigen::half>& input_diff_data,
Dstream.cc1318 const DeviceMemory<double> &input_diff_data, in ThenPoolBackward() argument
1323 PARAM(input_diff_data), PARAM(output_diff_data), in ThenPoolBackward()
1330 input_diff_data, output_diff_data, in ThenPoolBackward()
1348 const DeviceMemory<float> &input_diff_data, in ThenPoolBackward() argument
1353 PARAM(input_diff_data), PARAM(output_diff_data), in ThenPoolBackward()
1360 input_diff_data, output_diff_data, in ThenPoolBackward()
1375 const DeviceMemory<Eigen::half> &input_diff_data, in ThenPoolBackward() argument
1380 PARAM(input_diff_data), PARAM(output_diff_data), in ThenPoolBackward()
1387 input_diff_data, output_diff_data, in ThenPoolBackward()
Dstream.h550 const DeviceMemory<double> &input_diff_data,
559 const DeviceMemory<float> &input_diff_data,
568 const DeviceMemory<Eigen::half> &input_diff_data,