xla
Fix output tensor shape for argmin and argmax where keepdim=True and dim=None
#6536
Merged
Merged
Fix output tensor shape for argmin and argmax where keepdim=True and dim=None #6536
wonjoo-wj
merged
8 commits into
pytorch:master
from
master
Remove redundant log message when using IR debug, but not overriding …
57605e1b
Merge branch 'pytorch:master' into master
a65a8795
Fix the valid use case where dim=None and keepdim=True on torch.argmi…
4184bd9c
Merge branch 'pytorch:master' into master
6735ac11
wonjoo-wj
requested changes
on 2024-02-20
Linter fix
a8bbdb7e
Add C++ tests for argmin and argmax with dim=None and keepdim=true
97bd4dfb
wonjoo-wj
approved these changes
on 2024-02-27
Give tests unique names
1ac26223
Merge branch 'pytorch:master' into master
7a88949c
Login to write a write a comment.
Login via GitHub