No handlers found: "aten::einsum"

[787629504]

No handlers found: "aten::unfold" No handlers found: "aten::split" No handlers found: "aten::einsum". No handlers found: "aten::permute".

hope to support these ops, thanks!

[zhijian-liu]

Could you please provide an example model that contains aten::permute? Thank you!

[787629504]

Thank you for your reply!

x = torch.randn(2, 3, 5) x.size() torch.Size([2, 3, 5]) x.permute(2, 0, 1).size() torch.Size([5, 2, 3])

It seems to not require any multiplication and addition operations. I have calculated flops using other calculation codes, thank you!

[zhijian-liu]

Thanks! I was trying to refer to aten::einsum. Do you have an example of that?

[787629504]

The model in my code is a bit complex:

A = torch.randn(b,n,c,h,w,k) B = torch.randn(b,n,c,h,w,k) C = torch.einsum('bnchwk,bnchwk -> bnchw', A, B) C.shape torch.Size([b,n,c,h,w])

A simple example:

import torch A = torch.randn(3, 4) B = torch.randn(4, 5) C = torch.einsum('ik,kj->ij', A, B) C.shape torch.Size([3, 5])

Thank you!