IVGCVSW-2354 Caffe SqueezeNet through armnn Cl and Neon
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>
Thu, 14 Mar 2019 11:55:03 +0000 (11:55 +0000)
committerÁron Virginás-Tar <aron.virginas-tar@arm.com>
Fri, 15 Mar 2019 15:24:44 +0000 (15:24 +0000)
commit65d30965eef7e8534fc16ea4ded413c42a81c362
tree3ed591e57ee4bd68f98fce0d2b47aa0e621f2973
parent222c753ba83bec5dc915f01d999ff76826ec45d0
IVGCVSW-2354 Caffe SqueezeNet through armnn Cl and Neon

 * Compute Softmax Acl axis for Cl and Neon
 * Add unittests for Softmax in 3D and 4D
 * Correct input and output layer names in CaffeSqueezeNet inference test

Change-Id: I2d369d9a2db19c40f2af3341039dd33f0c5637b1
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
12 files changed:
src/backends/aclCommon/ArmComputeUtils.hpp
src/backends/backendsCommon/test/LayerTests.cpp
src/backends/backendsCommon/test/LayerTests.hpp
src/backends/backendsCommon/test/SoftmaxTestImpl.hpp
src/backends/cl/test/ClLayerTests.cpp
src/backends/cl/workloads/ClSoftmaxFloatWorkload.cpp
src/backends/cl/workloads/ClSoftmaxUint8Workload.cpp
src/backends/neon/test/NeonLayerTests.cpp
src/backends/neon/workloads/NeonSoftmaxFloatWorkload.cpp
src/backends/neon/workloads/NeonSoftmaxUint8Workload.cpp
src/backends/reference/test/RefLayerTests.cpp
tests/CaffeSqueezeNet1_0-Armnn/CaffeSqueezeNet1_0-Armnn.cpp