aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp')
-rw-r--r--src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp7
1 files changed, 4 insertions, 3 deletions
diff --git a/src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp b/src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp
index c45ab2cded..a0856a485b 100644
--- a/src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp
+++ b/src/armnn/optimizations/ConvertFp32NetworkToBf16.hpp
@@ -27,9 +27,10 @@ inline LayerT* ConvertWeight(Layer* l)
{
std::vector<BFloat16> newValues(info.GetNumElements());
- armnnUtils::FloatingPointConverter::ConvertFloat32ToBFloat16(layer->m_Weight->template GetTensor<float>(),
- info.GetNumElements(),
- newValues.data());
+ armnnUtils::FloatingPointConverter::ConvertFloat32ToBFloat16(
+ layer->m_Weight->template GetConstTensor<float>(),
+ info.GetNumElements(),
+ newValues.data());
TensorInfo newInfo(info);
newInfo.SetDataType(DataType::BFloat16);