Skip to content

[Relay] add ShapeFunc for one_hot op#7490

Merged
mbrookhart merged 3 commits into
apache:mainfrom
monklof:one_hot_shape_func
Mar 8, 2021
Merged

[Relay] add ShapeFunc for one_hot op#7490
mbrookhart merged 3 commits into
apache:mainfrom
monklof:one_hot_shape_func

Conversation

@monklof

@monklof monklof commented Feb 22, 2021

Copy link
Copy Markdown
Contributor

No description provided.

@monklof

monklof commented Feb 22, 2021

Copy link
Copy Markdown
Contributor Author

@kevinthesun @zhiics

@tqchen

tqchen commented Mar 3, 2021

Copy link
Copy Markdown
Member

ping @icemelon9 @mbrookhart @zhiics would be great if you can help to take a look.

@mbrookhart mbrookhart left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can you add a test that executes one_hot with a dynamic input shape so we hit this code?

@mbrookhart

Copy link
Copy Markdown
Contributor

Otherwise LGTM

@tqchen tqchen added the status: need test case need test cases to cover the change label Mar 3, 2021
@monklof

monklof commented Mar 4, 2021

Copy link
Copy Markdown
Contributor Author

@mbrookhart okay

@monklof

monklof commented Mar 8, 2021

Copy link
Copy Markdown
Contributor Author

Hi @mbrookhart Unit tests have been added, it's this what you want?

@icemelon icemelon left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@mbrookhart mbrookhart merged commit b827845 into apache:main Mar 8, 2021
@mbrookhart

Copy link
Copy Markdown
Contributor

Thanks @monklof @icemelon9 @tqchen

trevor-m pushed a commit to trevor-m/tvm that referenced this pull request May 6, 2021
* [Relay] add ShapeFunc for one_hot op

* fix pylint

* add test for shapefunc of one_hot op
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request May 11, 2021
* [Relay] add ShapeFunc for one_hot op

* fix pylint

* add test for shapefunc of one_hot op
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

status: need review status: need test case need test cases to cover the change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants