Skip to content

Commit d0c4692

Browse files
authored
Merge pull request #60 from InfiniTensor/dev_flatten
fix: add flatten front support
2 parents 416cd2e + 25d0c44 commit d0c4692

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/07onnx/src/operators.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
#include "operators/dynamic_quantize_linear.hh"
1313
#include "operators/einsum.hh"
1414
#include "operators/expand.hh"
15+
#include "operators/flatten.hh"
1516
#include "operators/gather.hh"
1617
#include "operators/gather_elements.hh"
1718
#include "operators/gemm.hh"
@@ -81,6 +82,7 @@ namespace refactor::onnx {
8182
REGISTER(ReduceSum , Reduce );
8283
REGISTER(ReduceSumSquare , Reduce );
8384
REGISTER(Reshape , Reshape );
85+
REGISTER(Flatten , Flatten );
8486
REGISTER(ScatterND , ScatterND );
8587
REGISTER(Max , Select );
8688
REGISTER(Min , Select );

0 commit comments

Comments
 (0)