From 8c2f651ad454d11dd36e212560af23dbea5b758e Mon Sep 17 00:00:00 2001 From: Narumol Prangnawarat Date: Fri, 20 Nov 2020 16:17:48 +0000 Subject: IVGCVSW-5538 Fix delegate DepthwiseConv2d, DIV, Reshape * Correct filter shape for DepthwiseConv2d * Remove non-support data type * Allow check for flatten on Reshape Signed-off-by: Narumol Prangnawarat Change-Id: Ic4be63d7c2f3a2b5e13a1530025a49464c21171b --- delegate/src/Redefine.hpp | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) (limited to 'delegate/src/Redefine.hpp') diff --git a/delegate/src/Redefine.hpp b/delegate/src/Redefine.hpp index fb58ffdf70..91295768d6 100644 --- a/delegate/src/Redefine.hpp +++ b/delegate/src/Redefine.hpp @@ -105,10 +105,13 @@ TfLiteStatus VisitReshapeOperator(DelegateData& delegateData, for (int i=0; i < reshapeOptions->num_dimensions; ++i) { targetShape.push_back(reshapeOptions->shape[i]); - elementCounter = elementCounter * reshapeOptions->shape[i]; + if (reshapeOptions->shape[i] > 0) + { + elementCounter = elementCounter * reshapeOptions->shape[i]; + } } // Check the number of elements match, otherwise fall back to using the second input tensor. - if (elementCounter == inputTensorInfo0.GetNumElements()) + if (elementCounter <= inputTensorInfo0.GetNumElements()) { targetShapeFound = true; } -- cgit v1.2.1