From 91471fc2f7ffa078e688ef830246c90953c2e0c5 Mon Sep 17 00:00:00 2001 From: Feiyue Chen Date: Wed, 29 Nov 2023 02:47:57 +0000 Subject: [PATCH] Refine pack mapper Rremove restrictions on data types of Pack op Type: Code Improvement Signed-off-by: Feiyue Chen --- op_map.cc | 11 ----------- 1 file changed, 11 deletions(-) diff --git a/op_map.cc b/op_map.cc index 7fdc32f..34aef02 100644 --- a/op_map.cc +++ b/op_map.cc @@ -3283,17 +3283,6 @@ struct LogicalOpMapper : public OpMapperBase { }; struct PackMapper : public OpMapperBase { - virtual bool IsOpSupported(TfLiteContext* context, - TfLiteNode* node, - const TfLiteRegistration* registration) const { - auto input_tensor = context->tensors[node->inputs->data[0]]; - if (input_tensor.type == kTfLiteInt32 || - (input_tensor.dims->size == 1 && (input_tensor.type == kTfLiteInt8 || - input_tensor.type == kTfLiteUInt8))) { - return false; - } - return true; - } bool HandleMapOp(vx::delegate::Delegate* delegate, std::vector>& inputs, std::vector>& outputs,