diff options
Diffstat (limited to 'src/runtime/CL/functions/CLConvertFullyConnectedWeights.cpp')
-rw-r--r-- | src/runtime/CL/functions/CLConvertFullyConnectedWeights.cpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/runtime/CL/functions/CLConvertFullyConnectedWeights.cpp b/src/runtime/CL/functions/CLConvertFullyConnectedWeights.cpp index 02927e83ad..68c0fb6ebf 100644 --- a/src/runtime/CL/functions/CLConvertFullyConnectedWeights.cpp +++ b/src/runtime/CL/functions/CLConvertFullyConnectedWeights.cpp @@ -28,8 +28,14 @@ namespace arm_compute void CLConvertFullyConnectedWeights::configure(const ICLTensor *input, ICLTensor *output, const TensorShape &original_input_shape, DataLayout data_layout) { + configure(CLKernelLibrary::get().get_compile_context(), input, output, original_input_shape, data_layout); +} + +void CLConvertFullyConnectedWeights::configure(const CLCompileContext &compile_context, const ICLTensor *input, ICLTensor *output, const TensorShape &original_input_shape, + DataLayout data_layout) +{ auto k = arm_compute::support::cpp14::make_unique<CLConvertFullyConnectedWeightsKernel>(); - k->configure(input, output, original_input_shape, data_layout); + k->configure(compile_context, input, output, original_input_shape, data_layout); _kernel = std::move(k); } |