I\'m trying to run BatchNormalization on a batch of ragged tensors in TF 2.x, but seem to run into errors while doing this. (I can do a conversion to and from ragged tensors bef