Move `sharding` to type domain in `traced_type_inner`

[avik-pal]

In traced_type_inner we only need sharding to compute the Sharding.ndevices. We should instead compute the devices before calling the function and then pass in the value as a Val.