diff options
Diffstat (limited to 'src/gpu/cl/operators/ClWinogradConv2d.cpp')
-rw-r--r-- | src/gpu/cl/operators/ClWinogradConv2d.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/gpu/cl/operators/ClWinogradConv2d.cpp b/src/gpu/cl/operators/ClWinogradConv2d.cpp index ffa1effc74..b4163a5986 100644 --- a/src/gpu/cl/operators/ClWinogradConv2d.cpp +++ b/src/gpu/cl/operators/ClWinogradConv2d.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2018-2021 Arm Limited. + * Copyright (c) 2018-2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -214,6 +214,7 @@ void ClWinogradConv2d::configure(const ClCompileContext &compile_context, ITenso (src->data_type() == DataType::F16))); // Configure output transform + _output_transform->set_target(CLScheduler::get().target()); _output_transform->configure(compile_context, &_batched_mm_output, biases, dst, winograd_info, act_info); _aux_mem = _batched_mm.workspace(); |