diff options
| author | Mike Kelly <mike.kelly@arm.com> | 2023-06-30 15:18:26 +0100 |
|---|---|---|
| committer | mike.kelly <mike.kelly@arm.com> | 2023-07-03 15:32:04 +0000 |
| commit | a9ac6ba643e8dc4fee88bd0e7e186f0918080c4b (patch) | |
| tree | 4e05fbb098b30f3d2c7d3e5b4b83d48c9bdd59ac /src/armnn/layers/LayerWithParameters.hpp | |
| parent | 6d2d4ead359aa02d502f15cfcb7e69c7658bd1ed (diff) | |
| download | armnn-a9ac6ba643e8dc4fee88bd0e7e186f0918080c4b.tar.gz | |
IVGCVSW-7828 Add an Optional TensorInfo to InputSlot
* Updated calls to use the new function
From:
GetInputSlot(n).GetConnection()->GetTensorInfo();
To:
GetInputSlot(n).GetTensorInfo();
* Added UnitTests
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I43184cc05e4472011b9347aaa820eb8deb1cd4a0
Diffstat (limited to 'src/armnn/layers/LayerWithParameters.hpp')
| -rw-r--r-- | src/armnn/layers/LayerWithParameters.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/layers/LayerWithParameters.hpp b/src/armnn/layers/LayerWithParameters.hpp index 0a1dbf363c..4e7cd45125 100644 --- a/src/armnn/layers/LayerWithParameters.hpp +++ b/src/armnn/layers/LayerWithParameters.hpp @@ -61,7 +61,7 @@ protected: Layer::ImmutableConstantTensors tensors; for (unsigned int i = 0; i < GetNumInputSlots(); ++i) { - if (GetInputSlot(i).GetConnection() && GetInputSlot(i).GetConnection()->GetTensorInfo().IsConstant()) + if (GetInputSlot(i).GetConnection() && GetInputSlot(i).GetTensorInfo().IsConstant()) { auto &inputLayer = GetInputSlot(i).GetConnectedOutputSlot()->GetOwningLayer(); if (inputLayer.GetType() == armnn::LayerType::Constant) |