Skip to content

[frontend][torch] Support aten::relu6 operator#12855

Merged
masahi merged 1 commit intoapache:mainfrom
chengven027:add.torch.relu6
Sep 21, 2022
Merged

[frontend][torch] Support aten::relu6 operator#12855
masahi merged 1 commit intoapache:mainfrom
chengven027:add.torch.relu6

Conversation

@chengven027
Copy link
Contributor

Hi, tvm:
When I use tvm pytorch frontend to load model vovnet19. I get a error: NotImplementedError: The following operators are not implemented: ['aten::relu6_'].
I tried adding an implementation of this operator,and then i success load my model and run success.
Hope to get your advice and review. Thanks.

@masahi
Copy link
Member

masahi commented Sep 21, 2022

It should be possible to support relu6 without introducing a new op. It's just a clip.

@chengven027
Copy link
Contributor Author

Thanks, I have already changed it. Please review again.

@masahi masahi merged commit 3c8a94b into apache:main Sep 21, 2022
@chengven027 chengven027 deleted the add.torch.relu6 branch September 22, 2022 09:42
xinetzone pushed a commit to daobook/tvm that referenced this pull request Nov 25, 2022
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

Successfully merging this pull request may close these issues.

2 participants