diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-09-10 17:20:34 +0100 |
|---|---|---|
| committer | Michalis Spyrou <michalis.spyrou@arm.com> | 2019-09-26 10:17:30 +0000 |
| commit | 1a569a30a2f456ff1a3e0a665201e1c3ab92df80 (patch) | |
| tree | 9d68934f461579edefbe65246f6ee435aaa18808 /src/graph/backends/NEON/NEFunctionFactory.cpp | |
| parent | f1cf394ae882e6e8fb2e0986f88d2548b82a85bb (diff) | |
| download | ComputeLibrary-1a569a30a2f456ff1a3e0a665201e1c3ab92df80.tar.gz | |
COMPMID-2161 [NEON] Create IWeightManager class
Change-Id: I1a9a46da2f98e896b825099151b56d1d8271dd31
Signed-off-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1915
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/graph/backends/NEON/NEFunctionFactory.cpp')
| -rw-r--r-- | src/graph/backends/NEON/NEFunctionFactory.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/graph/backends/NEON/NEFunctionFactory.cpp b/src/graph/backends/NEON/NEFunctionFactory.cpp index 852de549fa..45e9727133 100644 --- a/src/graph/backends/NEON/NEFunctionFactory.cpp +++ b/src/graph/backends/NEON/NEFunctionFactory.cpp @@ -115,6 +115,7 @@ std::unique_ptr<IFunction> create_convolution_layer<NEConvolutionLayerFunctions, std::shared_ptr<IMemoryManager> mm = get_memory_manager(ctx, Target::NEON); std::unique_ptr<IFunction> func; std::string func_name; + if(conv_algorithm == ConvolutionMethod::Direct) { std::tie(func, func_name) = create_named_memory_managed_function<NEDirectConvolutionLayer>( |