We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 416cd2e commit 25d0c44Copy full SHA for 25d0c44
src/07onnx/src/operators.cpp
@@ -12,6 +12,7 @@
12
#include "operators/dynamic_quantize_linear.hh"
13
#include "operators/einsum.hh"
14
#include "operators/expand.hh"
15
+#include "operators/flatten.hh"
16
#include "operators/gather.hh"
17
#include "operators/gather_elements.hh"
18
#include "operators/gemm.hh"
@@ -81,6 +82,7 @@ namespace refactor::onnx {
81
82
REGISTER(ReduceSum , Reduce );
83
REGISTER(ReduceSumSquare , Reduce );
84
REGISTER(Reshape , Reshape );
85
+ REGISTER(Flatten , Flatten );
86
REGISTER(ScatterND , ScatterND );
87
REGISTER(Max , Select );
88
REGISTER(Min , Select );
0 commit comments