onnx-mlir/SharingWork.md

Please keep up to date.

• As you start working on a new operation, move it from the lower table to the top table,
• write your name under 'person working on it,'
• once you have pushed the functionality in the master branch, mark it as complete.

ONNX to KRNL work.

Update as you push code to the master branch.

ONNX Operation	Person working on it	ONNX Def	Shape Inference	ONNX to KRNL	Optimized	Extended Functionality
Abs		v	v	v
Add		v	v	v		M
And		v	v	v		M
AveragePool	Tung	v	v	v		nP
BatchNorm		v	v	v
Constant		v	v	v
Conv		v	v	v		nP
Concat		v	v	v		V
Cos		v	v	v
Cosh		v	v	v
Div		v	v	v		M
Elu		v	v	v
Exp		v	v	v
Gemm		v	v	v		U
HardSigmoid		v	v	v
Identity		v	v	v
LeakyRelu		v	v	v
Log		v	v	v
MatMul		v	v	v		noM
Max		v	v	v		M V
MaxPool		v	v	v		M noIdx
Min		v	v	v		M V
Mul		v	v	v		noM
Or		v	v	v		M
Pad		v	V	v		const only
Reciprocal		v	v	v
ReduceMax		v	v	v
ReduceL1		v	v	v
ReduceL2		v	v	v
ReduceLogSum		v	v	v
ReduceMin		v	v	v
ReduceProd		v	v	v
ReduceSum		v	v	v
Relu		v	v	v
Reshape		v	v	v		const only
Selu		v	v	v
Sigmoid		v	v	v
Sign		v	v	v
Sinh		v	v	v
Softmax		v	v	v
Softplus		v	v	v
Softsign		v	v	v
Sqrt		v	v	v
Sub		v	v	v		M
Sum		v	v	v		M V
Tanh		v	v	v
Transpose		v	v	v
Unsqueeze		v	v	v
Xor		v	v	v		M

Extended functionality info (noXXX: additional work is needed).

• M / noM: multi-broadcast imlemented / not implemented
• U / noU: unidir-broadcast implemented / not implemented
• P / noP: padding implemented / not implemented
• V / noV: variadic input implemented / non implemented

Operation not supported yet.

Please pick new operations to work on, and make sure you utilize current code to minimize additional work needed to support new functionality.

Typical sequence of work.

1. Ensure the operation is properly parsed from ONNX and MLIR input.
2. Perform shape inference.
3. Lower to KRNL dialect.

And add literal tests at each step, and end to end tests once completed.

ONNX Operation	Person working on it	ONNX Def	Shape Inference	ONNX to KRNL	Basic Functionality	Extended functionality
ACos
ACosh
ArgMax
ArgMin
ASin
ASinh
ATan
ATanh
BitShift						M
Ceil
Celu
Clip
Compress
ConcatFromSeq
ConstOfShape
ConvInt						P
ConvTrans						P
CumSum
DepthToSpace
DequantizeLin
Det
Dropout
DynQuantizeLin
Einsum						V
Equal						M
Erf
Expand
EyeLike
Flatten
Floor
GRU
GatherElements
GatherND
GlobalAvgPool
GlobalLpPool
GlobalMaxPool
Greater						M
GreaterOrEq						M
HardMax
If						V
InstanceNorm
Inverse
IsInf
IsNaN
Less						M
LessOrEq						M
LSTM
LSTM
LSTM
LogSoftmax
Loop						V
LpNorm
LpPool
MatMulInteger						U
MaxRoiPool
MaxUnpool
Mean						M V
MeanSquareDist						M
MeanVarNorm
Mod						M
Multinomial
Neg
NegLogLikeLoss
NonMaxSupress
NonZero
Not
OneHot
PRelu						U
Power						M
QLinearConv						P
QLinearMatMul						M
QuantizeLinear
RNN
RandomNormal
RandNormalLike
RandUniform
RandUniformLike
Range
ReduceLogSumExp
ReduceMean
ReduceSumSquare
Resize
ReverseSequence
RoiAlign
Round
Scan						V
Scatter						deprecated for Scatter Element
ScatterElem
ScatterND
SequenceAt
SeqConstruct						V
SequenceErase
SequenceInsert
SequenceLength
Shape
Shrink
Slice
SoftMaxCross
SpaceToDepth
Split						V
SplitToSequence
Squeeze
StringNorm
Tan
TfldfVect
ThresholdRelu
Tile
TopK
UnfoldToDepth
Unique
Upsample						deprecated
Where						M