diff options
Diffstat (limited to 'ConversionUtils.hpp')
-rw-r--r-- | ConversionUtils.hpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/ConversionUtils.hpp b/ConversionUtils.hpp index 0fbd4e4d..5dc9993d 100644 --- a/ConversionUtils.hpp +++ b/ConversionUtils.hpp @@ -1451,14 +1451,16 @@ bool SetupAndTrackLayerOutputSlot(const HalOperation& operation, uint32_t outputIndex, armnn::IConnectableLayer& layer, const HalModel& model, - ConversionData& data) + ConversionData& data, + const armnn::TensorInfo* overrideOutputInfo = nullptr) { return SetupAndTrackLayerOutputSlot<HalPolicy>(operation, outputIndex, layer, outputIndex, model, - data); + data, + overrideOutputInfo); } template<typename HalPolicy, |