jiazhihao / TASO

The Tensor Algebra SuperOptimizer for Deep Learning
Apache License 2.0
687 stars 90 forks source link

Possible unsupported operators #15

Closed OuHangKresnik closed 4 years ago

OuHangKresnik commented 4 years ago

Here is the list of possible unsupported operators besides StridedSlice:

NonMaxSuppressionV2 Fill CropAndResize TopKV2 ResizeNearestNeighbor, Merge

I see there is TopK in onnx while don't know what's the difference yet, investigating now. For this operator plz see: https://github.com/onnx/tensorflow-onnx/issues/715 That's why I convert it as custom operator instead of the official conversion. So does ResizesNearestNeighbor, Fill

OuHangKresnik commented 4 years ago

How about pay attention to NonMaxSuppressionV2, Merge and CropAndResize first. I will continue to investigate TopKV2, ResizesNearestNeighbor and Fill

OuHangKresnik commented 4 years ago

TopKV2 , ResizesNearestNeighbor and Fill can be successfully transformed to onnx operators if upgraded the opset. So does the StridedSlice

interesaaat commented 4 years ago

I would also add gather if possible.