diff --git a/aten/src/ATen/native/native_functions.yaml b/aten/src/ATen/native/native_functions.yaml index 7c9d819cfcf..5fb815f3c5b 100644 --- a/aten/src/ATen/native/native_functions.yaml +++ b/aten/src/ATen/native/native_functions.yaml @@ -3855,7 +3855,7 @@ precomputed: - dim -> int dim dispatch: - CPU, CUDA: max_out + CPU, CUDA, MTIA: max_out MPS: max_out_mps - func: max.names_dim(Tensor self, Dimname dim, bool keepdim=False) -> (Tensor values, Tensor indices) @@ -4044,7 +4044,7 @@ precomputed: - dim -> int dim dispatch: - CPU, CUDA: min_out + CPU, CUDA, MTIA: min_out MPS: min_out_mps - func: min.names_dim(Tensor self, Dimname dim, bool keepdim=False) -> (Tensor values, Tensor indices)