diff options
author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2018-03-21 17:45:31 +0000 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:49:16 +0000 |
commit | ed99f411d52949720a4d64d91664cd71e46b79d5 (patch) | |
tree | d903b523dea830aeb48d59a66b8da59e4dcf707a /src/graph2/backends | |
parent | 6528aa20e768f2d801328aa164d672b7fdfe266f (diff) | |
download | ComputeLibrary-ed99f411d52949720a4d64d91664cd71e46b79d5.tar.gz |
COMPMID-1018 - Add Winograd support in VGG16 and Alexnet examples
Change-Id: I4a2deee9e4b2c54ea79d2895cfeca44190133b24
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/125453
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/graph2/backends')
-rw-r--r-- | src/graph2/backends/CL/CLFunctionsFactory.cpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/graph2/backends/CL/CLFunctionsFactory.cpp b/src/graph2/backends/CL/CLFunctionsFactory.cpp index bba0cce025..5a51b19e18 100644 --- a/src/graph2/backends/CL/CLFunctionsFactory.cpp +++ b/src/graph2/backends/CL/CLFunctionsFactory.cpp @@ -165,7 +165,13 @@ std::unique_ptr<IFunction> create_convolution_layer(ConvolutionLayerNode &node, std::shared_ptr<IMemoryManager> mm = get_memory_manager(ctx, Target::CL); std::unique_ptr<IFunction> func; std::string func_name; - if(conv_algorithm == ConvolutionMethod::DIRECT) + + if(conv_algorithm == ConvolutionMethod::WINOGRAD) + { + std::tie(func, func_name) = create_named_function<CLWinogradConvolutionLayer>( + std::string("CLWinogradConvolutionLayer"), input, weights, biases, output, conv_info); + } + else if(conv_algorithm == ConvolutionMethod::DIRECT) { std::tie(func, func_name) = create_named_function<CLDirectConvolutionLayer>( std::string("CLDirectConvolutionLayer"), input, weights, biases, output, conv_info); |