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
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

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