[tf.data] Replacing usage of `std::unique_ptr<Foo>(new Foo(...))` with...

[tf.data] Replacing usage of `std::unique_ptr<Foo>(new Foo(...))` with `absl::make_unique<Foo>(...)` when possible and wrapping it `absl::WrapUnique<Foo>` otherwise.

PiperOrigin-RevId: 228961343