aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/CL/functions/CLDirectConvolutionLayer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/runtime/CL/functions/CLDirectConvolutionLayer.cpp')
-rw-r--r--src/runtime/CL/functions/CLDirectConvolutionLayer.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/CL/functions/CLDirectConvolutionLayer.cpp b/src/runtime/CL/functions/CLDirectConvolutionLayer.cpp
index 759692505b..50b7b46d99 100644
--- a/src/runtime/CL/functions/CLDirectConvolutionLayer.cpp
+++ b/src/runtime/CL/functions/CLDirectConvolutionLayer.cpp
@@ -56,7 +56,7 @@ void CLDirectConvolutionLayer::configure(ICLTensor *input, const ICLTensor *weig
Error CLDirectConvolutionLayer::validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, const PadStrideInfo &conv_info)
{
- return CLDirectConvolutionLayerKernel::validate(input, weights, biases, output, conv_info);
+ return CLDirectConvolutionLayerKernel::validate(input, weights, biases, output, conv_info, CLScheduler::get().target());
}
void CLDirectConvolutionLayer::run()