From 1628af434a0d9dbd24f484c98aa5ff895b335cd5 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/On-Device=20Lab=28SR=29/Princip?= =?utf8?q?al=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Fri, 9 Aug 2019 12:38:58 +0900 Subject: [PATCH] [moco-tf] Check attributes for ConcatV2 validate (#6428) This will add attribute existence check for ConcatV2 node validation Signed-off-by: SaeHie Park --- compiler/moco-tf/src/Op/Concat.cpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/compiler/moco-tf/src/Op/Concat.cpp b/compiler/moco-tf/src/Op/Concat.cpp index 7f81cce..7008f69 100644 --- a/compiler/moco-tf/src/Op/Concat.cpp +++ b/compiler/moco-tf/src/Op/Concat.cpp @@ -117,6 +117,9 @@ namespace tf bool ConcatV2GraphBuilderBase::validate(const tensorflow::NodeDef &node) const { + if (!plier::tf::has_attrs(node, {"T", "N", "Tidx"})) + return false; + // Concat node SHOULD have 3 or more inputs, that is 2 + axis const int num_inputs = node.input_size() - 1; assert(num_inputs >= 2); -- 2.7.4