| 5476 | } |
| 5477 | |
| 5478 | bool CudnnSupport::DoFusedConvolve( |
| 5479 | Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, |
| 5480 | const DeviceMemory<double>& conv_input_data, double conv_input_scale, |
| 5481 | const dnn::FilterDescriptor& filter_descriptor, |
| 5482 | const DeviceMemory<double>& filter_data, |
| 5483 | const dnn::ConvolutionDescriptor& convolution_descriptor, |
| 5484 | const DeviceMemory<double>& side_input_data, double side_input_scale, |
| 5485 | const dnn::BatchDescriptor& bias_descriptor, |
| 5486 | const DeviceMemory<double>& biases, dnn::ActivationMode activation_mode, |
| 5487 | const dnn::BatchDescriptor& output_descriptor, |
| 5488 | DeviceMemory<double>* output_data, ScratchAllocator* scratch_allocator, |
| 5489 | const dnn::AlgorithmConfig& algorithm_config, |
| 5490 | dnn::ProfileResult* output_profile_result) { |
| 5491 | return IsStatusOk( |
| 5492 | DoFusedConvolveImpl( |
| 5493 | stream, conv_input_descriptor, conv_input_data, conv_input_scale, |
| 5494 | filter_descriptor, filter_data, convolution_descriptor, |
| 5495 | side_input_data, side_input_scale, bias_descriptor, biases, |
| 5496 | activation_mode, output_descriptor, output_data, |
| 5497 | GetConvAccumulatorType(dnn::DataType::kDouble), scratch_allocator, |
| 5498 | algorithm_config, output_profile_result), |
| 5499 | /*report_error=*/!output_profile_result); |
| 5500 | } |
| 5501 | |
| 5502 | bool CudnnSupport::DoFusedConvolve( |
| 5503 | Stream* stream, const dnn::BatchDescriptor& conv_input_descriptor, |
nothing calls this directly
no test coverage detected