diff options
Diffstat (limited to 'shim/sl/canonical/ConversionUtils.hpp')
-rw-r--r-- | shim/sl/canonical/ConversionUtils.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/shim/sl/canonical/ConversionUtils.hpp b/shim/sl/canonical/ConversionUtils.hpp index 8058bcb379..beee00d11a 100644 --- a/shim/sl/canonical/ConversionUtils.hpp +++ b/shim/sl/canonical/ConversionUtils.hpp @@ -79,6 +79,8 @@ public: void SanitizeQuantizationScale(LayerInputHandle& weight, LayerInputHandle& input); + armnn::IOutputSlot* GetOutputSlot() const; + private: armnn::IOutputSlot* m_OutputSlot; bool m_Valid; @@ -1012,4 +1014,6 @@ ConstTensorPin DequantizeAndMakeConstTensorPin(const Operation& operation, size_t operandIndex, bool optional = false); +bool IsConnectedToDequantize(armnn::IOutputSlot* ioutputSlot); + } // namespace armnn_driver |