diff options
Diffstat (limited to 'src/armnn/optimizations/AddBroadcastReshapeLayer.hpp')
| -rw-r--r-- | src/armnn/optimizations/AddBroadcastReshapeLayer.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp b/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp index 26661cfcde..0a5ad9d152 100644 --- a/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp +++ b/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp @@ -74,7 +74,7 @@ public: ConstantLayer& constantLayer = static_cast<ConstantLayer&>(parentLayer); constantLayer.m_LayerOutput = std::make_unique<ScopedCpuTensorHandle>( - ConstTensor(reshapeInfo,constantLayer.m_LayerOutput.get()->GetTensor<void>())); + ConstTensor(reshapeInfo,constantLayer.m_LayerOutput.get()->GetConstTensor<void>())); constantLayer.GetOutputSlot().SetTensorInfo(reshapeInfo); return; |