Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h500 DeviceMemory<double>* output_diff_data,
510 DeviceMemory<float>* output_diff_data,
520 DeviceMemory<Eigen::half>* output_diff_data,
894 DeviceMemory<T>* output_diff_data,
Drocm_dnn.cc4198 DeviceMemory<T>* output_diff_data, ScratchAllocator* workspace_allocator) { in DoPoolBackwardImpl() argument
4292 output_diff_data->opaque(), workspace_ptr); in DoPoolBackwardImpl()
4310 DeviceMemory<double>* output_diff_data, in DoPoolBackward() argument
4323 DeviceMemory<float>* output_diff_data, in DoPoolBackward() argument
4327 input_diff_data, output_diff_data, in DoPoolBackward()
4338 DeviceMemory<Eigen::half>* output_diff_data, in DoPoolBackward() argument
4342 input_diff_data, output_diff_data, in DoPoolBackward()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h484 DeviceMemory<double>* output_diff_data,
494 DeviceMemory<float>* output_diff_data,
504 DeviceMemory<Eigen::half>* output_diff_data,
Dcuda_dnn.cc4368 DeviceMemory<double>* output_diff_data, in DoPoolBackward() argument
4385 output_diff_data->opaque())); in DoPoolBackward()
4398 DeviceMemory<float>* output_diff_data, in DoPoolBackward() argument
4415 output_diff_data->opaque())); in DoPoolBackward()
4428 DeviceMemory<Eigen::half>* output_diff_data, in DoPoolBackward() argument
4445 output_diff_data->opaque())); in DoPoolBackward()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1685 DeviceMemory<double>* output_diff_data, in DoPoolBackward() argument
1698 DeviceMemory<float>* output_diff_data, in DoPoolBackward() argument
1711 DeviceMemory<Eigen::half>* output_diff_data, in DoPoolBackward() argument
Dstream.cc909 DeviceMemory<double> *output_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()
936 DeviceMemory<float> *output_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()
961 DeviceMemory<Eigen::half> *output_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.h574 DeviceMemory<double> *output_diff_data,
583 DeviceMemory<float> *output_diff_data,
592 DeviceMemory<Eigen::half> *output_diff_data,