Calculation on tensor views fails on MPS - Placeholder shape mismatches

[gau-nernst]

🐛 Describe the bug

Calculation on tensor views fails on MPS. Python crashes

import torch

x = torch.randn(2, 3, 3, device="mps")
x = x[:1]
out = x[:, 0:1, 0:1] * x[:, 1:2, 1:2]

Output

/AppleInternal/Library/BuildRoots/810eba08-405a-11ed-86e9-6af958a02716/Library/Caches/com.apple.xbs/Sources/MetalPerformanceShadersGraph/mpsgraph/MetalPerformanceShadersGraph/Core/Files/MPSGraph.mm:673: failed assertion `Placeholder shape mismatches (expected 1 vs got tensorData with 3) at dimIdx = 2'
[1]    73020 abort      python debug.py

There is no crash when I change device="cpu" or remove x = x[:1]

Versions

Collecting environment information...
PyTorch version: 2.0.0.dev20230224
Is debug build: False
CUDA used to build PyTorch: None
ROCM used to build PyTorch: N/A

OS: macOS 13.0 (arm64)
GCC version: Could not collect
Clang version: 14.0.0 (clang-1400.0.29.102)
CMake version: version 3.24.0
Libc version: N/A

Python version: 3.10.9 | packaged by conda-forge | (main, Feb 2 2023, 20:26:08) [Clang 14.0.6 ] (64-bit runtime)
Python platform: macOS-13.0-arm64-arm-64bit
Is CUDA available: False
CUDA runtime version: No CUDA
CUDA_MODULE_LOADING set to: N/A
GPU models and configuration: No CUDA
Nvidia driver version: No CUDA
cuDNN version: No CUDA
HIP runtime version: N/A
MIOpen runtime version: N/A
Is XNNPACK available: True

CPU:
Apple M1

Versions of relevant libraries:
[pip3] mypy==1.0.1
[pip3] mypy-extensions==1.0.0
[pip3] numpy==1.24.2
[pip3] torch==2.0.0.dev20230224
[conda] numpy 1.24.2 pypi_0 pypi
[conda] pytorch 2.0.0.dev20230224 py3.10_0 pytorch-nightly

cc @kulinseth @albanD @malfet @DenisVieriu97 @razarmehr @abhudev