diff options
Diffstat (limited to 'src/graph/nodes/DepthwiseConvolutionLayer.cpp')
-rw-r--r-- | src/graph/nodes/DepthwiseConvolutionLayer.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/graph/nodes/DepthwiseConvolutionLayer.cpp b/src/graph/nodes/DepthwiseConvolutionLayer.cpp index ceac2a2def..b4598538c1 100644 --- a/src/graph/nodes/DepthwiseConvolutionLayer.cpp +++ b/src/graph/nodes/DepthwiseConvolutionLayer.cpp @@ -72,6 +72,9 @@ std::unique_ptr<arm_compute::IFunction> DepthwiseConvolutionLayer::instantiate_n node_ctx.add_parameter<PadStrideInfo>("ConvolutionInfo", _conv_info); node_ctx.add_parameter<bool>("Optimized3x3", _opt3x3); + // Configure operation + auto func = OperationRegistry::get().find_operation(OperationType::DepthwiseConvolutionLayer, _target_hint)->configure(node_ctx); + // Fill tensors if(!weights_is_loaded) { @@ -83,5 +86,5 @@ std::unique_ptr<arm_compute::IFunction> DepthwiseConvolutionLayer::instantiate_n } // Get function - return OperationRegistry::get().find_operation(OperationType::DepthwiseConvolutionLayer, _target_hint)->configure(node_ctx); + return func; } |