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

Shape not implemented yet #28

Closed
Ontheway361 opened this issue Dec 5, 2022 · 1 comment
Closed

Shape not implemented yet #28

Ontheway361 opened this issue Dec 5, 2022 · 1 comment

Comments

@Ontheway361
Copy link

PyTorch code:
def forward(self, x):
## x.shape = [B, T, F]
B, _, _ = x.shape
x = x.unsqueeze(1).transpose(2, 3)
x = self.layers(x)
x = torch.mean(x, dim=2, keepdim=True)
x = x.view((B, self.last_channel, -1))
x = x.transpose(1, 2)
return x

when convert onnx model to tflite, builder.py throw a KeyError, maybe we need to implement Shape op.

@MPolaris
Copy link
Owner

MPolaris commented Dec 5, 2022

Hi Ontheway361, you can implement it by yourself at first, and welcome PR.

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