Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

These 9op are not supported:{Shape,Gather,Unsqueeze,Equal,Cast,TopK,Expand,OneHot,ReduceSum,} #231

Open
fyq00 opened this issue Mar 24, 2020 · 1 comment

Comments

@fyq00
Copy link

fyq00 commented Mar 24, 2020

I know 3op : Shape, Gather, Unsqueeze can be change by simplifier onnx too.
But here are still 6op I need.
My model need to use heatmap prediction. I use function of pytorch :
1 torch.topk() to sort pixel scores.
2 OPtion : (Tensor1 == Tensor2) to zero the other unimportant pixels ....
3 torch.expand() function ~
and so on ~

But it's seem like Tengine still not support this op.

I dont know how to solve this problem~
Can you help me ~

@liqi-c
Copy link
Collaborator

liqi-c commented Mar 31, 2020

topk we support now , but maybe not support in onnx format , Maybe you can try convert you onnx model to tengine models format , and then we can know which op is not support , We will add it later
you alse can reference the tools/plugin/serializer/onnx/onnx_serializer.cpp to check which op we support .

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants