From a4983cec09a3e24bf4e99abd31aa11842e8b365f Mon Sep 17 00:00:00 2001 From: Finn Williams Date: Thu, 23 Jul 2020 12:55:12 +0100 Subject: 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 Signed-off-by: Teresa Charlin --- Utils.hpp | 3 +++ 1 file changed, 3 insertions(+) (limited to 'Utils.hpp') diff --git a/Utils.hpp b/Utils.hpp index d58d2735..86eb6db4 100644 --- a/Utils.hpp +++ b/Utils.hpp @@ -143,6 +143,9 @@ void RenameGraphDotFile(const std::string& oldName, const std::string& dumpDir, /// Checks if a tensor info represents a dynamic tensor bool IsDynamicTensor(const armnn::TensorInfo& outputInfo); +/// Checks for ArmNN support of dynamic tensors. +bool AreDynamicTensorsSupported(void); + std::string GetFileTimestamp(); #if defined(ARMNN_ANDROID_NN_V1_2) || defined(ARMNN_ANDROID_NN_V1_3) -- cgit v1.2.1