From dca769b9673a1e197258f7b35637b2a17f1a9e8b Mon Sep 17 00:00:00 2001 From: James Conroy Date: Tue, 27 Apr 2021 17:13:27 +0100 Subject: IVGCVSW-5815 Generalise ConstCpuTensorHandle * Generalises ConstCpuTensorHandle and inherited classes by removing 'Cpu' from aliases. * New renamed classes: ConstTensorHandle, TensorHandle, ScopedTensorHandle, PassthroughTensorHandle, ConstPassthroughTensorHandle. Signed-off-by: James Conroy Change-Id: I1824e0e134202735fb77051f20a7252f161dfe16 --- src/armnn/optimizations/AddBroadcastReshapeLayer.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src/armnn/optimizations/AddBroadcastReshapeLayer.hpp') diff --git a/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp b/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp index aa00b9913c..4cfe2e4898 100644 --- a/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp +++ b/src/armnn/optimizations/AddBroadcastReshapeLayer.hpp @@ -8,7 +8,7 @@ #include #include -#include +#include namespace armnn { @@ -70,7 +70,7 @@ public: { ConstantLayer& constantLayer = static_cast(parentLayer); - constantLayer.m_LayerOutput = std::make_unique( + constantLayer.m_LayerOutput = std::make_unique( ConstTensor(reshapeInfo, constantLayer.m_LayerOutput.get()->GetConstTensor())); constantLayer.GetOutputSlot().SetTensorInfo(reshapeInfo); } -- cgit v1.2.1