diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-01-20 15:58:29 +0000 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-01-25 17:56:59 +0000 |
commit | 0b51d5ad533f8ecde71f957077690195eea29ffc (patch) | |
tree | d04aaecd63deb8c67f4cea001bc4ddac3181911c /src/armnn/optimizations | |
parent | e5617954db782628ca49919a627d01ee0088fb67 (diff) | |
download | armnn-0b51d5ad533f8ecde71f957077690195eea29ffc.tar.gz |
IVGCVSW-5619 Add OptimizerOptions and NetworkProperties to ArmNN Delegate
* Add OptimizerOptions, NetworkProperties, DebugCallbackFunction
to DelegateOptions
* Enable OptimizerOptions when the network is being optimized
* Enable NetworkProperties when loading network
* Enable DebugCallbackFunction
* Add error message when loading network
* Log warning instead of error when operator is not supported but
could fallback to another backend
* Improve uint16_t CompareData
* Unit tests
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I353035afb442774bfeb1c62570a90755c2ceaf38
Diffstat (limited to 'src/armnn/optimizations')
-rw-r--r-- | src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp b/src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp index ca42cacb39..c45ab2cded 100644 --- a/src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp +++ b/src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp @@ -31,7 +31,8 @@ inline LayerT* ConvertWeight(Layer* l) info.GetNumElements(), newValues.data()); - TensorInfo newInfo(info.GetShape(), DataType::BFloat16); + TensorInfo newInfo(info); + newInfo.SetDataType(DataType::BFloat16); ConstTensor newInput(newInfo, newValues); layer->m_Weight.reset(new ScopedCpuTensorHandle(newInput)); } |