Remove control dependency escape from PartitionedVariable.concat().

Wrapping the as_tensor() method of PartitionedVariable is incorrect. It means, for example, that concatenation happens outside the context of any enclosing loop, so loop iterations would see stale values. In particular this is the wrong behavior for models wrapped in a in-graph training loop.

PiperOrigin-RevId: 208663593