Use TensorBase in Shuffle and WeightNorm cpu kernels (#84499)
These files are already only using the subset avalilable with
TensorBase, so this is straight-forward name substitution.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/84499
Approved by: https://github.com/ezyang