Refine pack mapper

Rremove restrictions on data types of Pack op Type: Code Improvement Signed-off-by: Feiyue Chen <[email protected]>
VeriSilicon · Dec 26, 2023 · 91471fc · 91471fc
1 parent d8552f8
commit 91471fc
Showing 1 changed file with 0 additions and 11 deletions.
diff --git a/op_map.cc b/op_map.cc
@@ -3283,17 +3283,6 @@ struct LogicalOpMapper : public OpMapperBase<EmptyStructPlaceholder> {
 };
 
 struct PackMapper : public OpMapperBase<TfLitePackParams> {
-  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<std::shared_ptr<tim::vx::Tensor>>& inputs,
                    std::vector<std::shared_ptr<tim::vx::Tensor>>& outputs,