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

Gather operation with indices as tensor expr in TFLite frontend #6168

Merged
merged 2 commits into from
Aug 18, 2020

Conversation

d-smirnov
Copy link
Contributor

This PR adds handling of indices as tensor expr to "gather" operation in TFLite frontend. Also the code checking out of boundary error refactored in slightly more "pythonic" way. Fixed bug in negative axis value normalisation

Copy link
Member

@siju-samuel siju-samuel left a comment

Choose a reason for hiding this comment

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

@d-smirnov LGTM, just one small nit.

python/tvm/relay/frontend/tflite.py Outdated Show resolved Hide resolved
Added handling of indices as tensor expr
to gather operation, unit tests amended
Code cheking out of boundary error refactored
in more "pythonic" way. Fixed bug in negative
axis value normalisation
@d-smirnov d-smirnov force-pushed the gather_indices_expr branch from c1df2be to 2bd509f Compare August 13, 2020 11:33
Copy link
Member

@siju-samuel siju-samuel left a comment

Choose a reason for hiding this comment

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

LGTM

@d-smirnov d-smirnov requested a review from siju-samuel August 13, 2020 15:48
Copy link
Contributor

@anijain2305 anijain2305 left a comment

Choose a reason for hiding this comment

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

LGTM

@anijain2305 anijain2305 merged commit 7704232 into apache:master Aug 18, 2020
@anijain2305
Copy link
Contributor

Thanks @siju-samuel @d-smirnov This is merged!

trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Aug 26, 2020
…he#6168)

* gather with indices as tensor expr

Added handling of indices as tensor expr
to gather operation, unit tests amended
Code cheking out of boundary error refactored
in more "pythonic" way. Fixed bug in negative
axis value normalisation

* replaced with get_tensor_expr
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Aug 26, 2020
…he#6168)

* gather with indices as tensor expr

Added handling of indices as tensor expr
to gather operation, unit tests amended
Code cheking out of boundary error refactored
in more "pythonic" way. Fixed bug in negative
axis value normalisation

* replaced with get_tensor_expr
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Aug 26, 2020
…he#6168)

* gather with indices as tensor expr

Added handling of indices as tensor expr
to gather operation, unit tests amended
Code cheking out of boundary error refactored
in more "pythonic" way. Fixed bug in negative
axis value normalisation

* replaced with get_tensor_expr
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Sep 2, 2020
…he#6168)

* gather with indices as tensor expr

Added handling of indices as tensor expr
to gather operation, unit tests amended
Code cheking out of boundary error refactored
in more "pythonic" way. Fixed bug in negative
axis value normalisation

* replaced with get_tensor_expr
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Sep 3, 2020
…he#6168)

* gather with indices as tensor expr

Added handling of indices as tensor expr
to gather operation, unit tests amended
Code cheking out of boundary error refactored
in more "pythonic" way. Fixed bug in negative
axis value normalisation

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

Successfully merging this pull request may close these issues.

4 participants