diff options
| author | Finn Williams <Finn.Williams@arm.com> | 2020-07-23 12:55:12 +0100 |
|---|---|---|
| committer | Teresa Charlin <teresa.charlinreyes@arm.com> | 2020-08-12 17:06:37 +0100 |
| commit | a4983cec09a3e24bf4e99abd31aa11842e8b365f (patch) | |
| tree | 0a09be7cdad7ad5c15bb24c67abf0f4cd4d6e47e /ConversionUtils.cpp | |
| parent | bf866e2dc7bd5936788fe213b5c0f74483ec1532 (diff) | |
| download | android-nn-driver-a4983cec09a3e24bf4e99abd31aa11842e8b365f.tar.gz | |
IVGCVSW-4931 Update NN Driver to support dynamic tensors
* Change NN Driver m_Network to now have ShapeInferenceMethod::InferAndValidate
* Implement dynamic tensor support for:
- ArgMinMax layer
- Pooling2d layer
- Activation layer
* Skip dynamic tensor tests for any HAL other than 1.3
Change-Id: Icf66c968e49cdd4822b8c79c5f18b3f9e97dc53f
Signed-off-by: Finn Williams <Finn.Williams@Arm.com>
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Diffstat (limited to 'ConversionUtils.cpp')
| -rw-r--r-- | ConversionUtils.cpp | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/ConversionUtils.cpp b/ConversionUtils.cpp index 4c773964..b03ffbda 100644 --- a/ConversionUtils.cpp +++ b/ConversionUtils.cpp @@ -38,6 +38,15 @@ void LayerInputHandle::Connect(armnn::IInputSlot& inputSlot) } } +void LayerInputHandle::Disconnect(armnn::IInputSlot& inputSlot) +{ + ARMNN_ASSERT(IsValid()); + if (m_OutputSlot) + { + m_OutputSlot->Disconnect(inputSlot); + } +} + const armnn::TensorInfo& LayerInputHandle::GetTensorInfo() const { return m_TensorInfo; |