[ATen] Use MaybeOwned<T> in at::argmin/argmax (#58338)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58338
Test Plan: CI
Reviewed By: swolchok
Differential Revision: D28458968
fbshipit-source-id: 2c759bdb9fbdbef32d804f6d8efb09fb1d2bb30a