From 25d0c443ca7ec75804adb2c488421f76b207ce98 Mon Sep 17 00:00:00 2001
From: zhangyunze <z13785159769@163.com>
Date: Wed, 27 Dec 2023 14:47:56 +0800
Subject: [PATCH] fix: add flatten front support

---
 src/07onnx/src/operators.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/07onnx/src/operators.cpp b/src/07onnx/src/operators.cpp
index 2fbfe8b0..6e657f27 100644
--- a/src/07onnx/src/operators.cpp
+++ b/src/07onnx/src/operators.cpp
@@ -12,6 +12,7 @@
 #include "operators/dynamic_quantize_linear.hh"
 #include "operators/einsum.hh"
 #include "operators/expand.hh"
+#include "operators/flatten.hh"
 #include "operators/gather.hh"
 #include "operators/gather_elements.hh"
 #include "operators/gemm.hh"
@@ -81,6 +82,7 @@ namespace refactor::onnx {
         REGISTER(ReduceSum            , Reduce               );
         REGISTER(ReduceSumSquare      , Reduce               );
         REGISTER(Reshape              , Reshape              );
+        REGISTER(Flatten              , Flatten              );
         REGISTER(ScatterND            , ScatterND            );
         REGISTER(Max                  , Select               );
         REGISTER(Min                  , Select               );