xla
implement Repeat with fixed output shape
#7114
Merged

implement Repeat with fixed output shape #7114

alanwaketan merged 2 commits into master from JackCaoG/repeat_static
JackCaoG
JackCaoG torch_xla/experimental/custom_kernel.py
ac79b2a1
JackCaoG JackCaoG changed the title torch_xla/experimental/custom_kernel.py implement Repeat with fixed output shape 1 year ago
JackCaoG add test
736d0c72
JackCaoG JackCaoG marked this pull request as ready for review 1 year ago
JackCaoG JackCaoG requested a review from alanwaketan alanwaketan 1 year ago
alanwaketan
alanwaketan commented on 2024-05-25
alanwaketan
alanwaketan approved these changes on 2024-05-25
alanwaketan
alanwaketan alanwaketan merged 3369bf7d into master 1 year ago
alanwaketan alanwaketan deleted the JackCaoG/repeat_static branch 1 year ago
alanwaketan
alanwaketan alanwaketan requested a review from miladm miladm 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone