projects / platform / upstream / armnn.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: cf2ad55) | patch
IVGCVSW-4520 Implement BFloat16 Optimizer
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>
Fri, 20 Mar 2020 15:01:01 +0000 (15:01 +0000)
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>
Fri, 20 Mar 2020 19:09:07 +0000 (19:09 +0000)
commitbc7ffb5e9e5f4c86280b20c65772eb12d8bb140e
tree5187f34326414e7dfea80e0f4efaae5cbeb05d1dtree | snapshot
parentcf2ad554502830804e991aca2e5b0741623119b2commit | diff
IVGCVSW-4520 Implement BFloat16 Optimizer

 * Add ReduceFp32ToBf16 to OptimizerOptions
 * Add ConvertFp32NetworkToBf16
 * Add utility functions to insert conversion layers
 * Add constant conversion BF16 <-> FP32
 * Unit tests

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Iaff77e20c721400b052cb37eb9ef6fe16d7abaff
12 files changed:
Android.mk diff | blob | history
CMakeLists.txt diff | blob | history
include/armnn/INetwork.hpp diff | blob | history
src/armnn/CompatibleTypes.hpp diff | blob | history
src/armnn/Network.cpp diff | blob | history
src/armnn/NetworkUtils.cpp diff | blob | history
src/armnn/NetworkUtils.hpp diff | blob | history
src/armnn/optimizations/All.hpp diff | blob | history
src/armnn/optimizations/ConvertConstants.hpp diff | blob | history
src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp [new file with mode: 0644] blob
src/armnn/test/optimizations/ConvertConstantsBFloatTests.cpp [new file with mode: 0644] blob
src/armnn/test/optimizations/Fp32NetworkToBf16ConverterTests.cpp [new file with mode: 0644] blob
Domain: Machine Learning / ML Framework;