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

Bert Slice Support #12730

Closed
ntarafdar opened this issue Sep 16, 2024 · 0 comments
Closed

Bert Slice Support #12730

ntarafdar opened this issue Sep 16, 2024 · 0 comments
Assignees

Comments

@ntarafdar
Copy link
Contributor

ntarafdar commented Sep 16, 2024

I accidentally made the issue in the wrong project (pytorch2.0) , so I'm copying it here.

Sprint on making sure all slice configs are supported in Bert.

These are the shapes, I think they should be supported but please can we check and report any issues.
Update the Status column for the shapes that are supported and add these shapes to unit test.

As a side also make sure these shapes are swept on as we already have a sweep test for slice

CC: @ayerofieiev-tt

aten.slice.Tensor

ATen Input Variations Status
0 Tensor<[1, 1, 1, 256]> self = ?,
int<> dim = 3,
Optional[int]<> start = 0,
Optional[int]<> end = -1
Unknown
1 Tensor<[1, 256]> self = ?,
int<> dim = 0,
Optional[int]<> start = 0,
Optional[int]<> end = -1
Unknown
2 Tensor<[1, 512]> self = ?,
int<> dim = 0,
Optional[int]<> start = 0,
Optional[int]<> end = -1
Unknown
3 Tensor<[1, 512]> self = ?,
int<> dim = 1,
Optional[int]<> start = 0,
Optional[int]<> end = 256
Unknown
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants