expose tfrecord dataset transformation function for LinkedIn usage

[YazhiGao]

Description

For LinkedIn internal usage, expose the tfrecord dataset transformation functionality as a standalone function.

Fixes # (issue)

Type of change

Please delete options that are not relevant.

• [x] New feature (non-breaking change which adds functionality)

List all changes

Please list all changes in the commit.

• add new function tfrecord_transform_fn

Testing

Please describe the tests that you ran to verify your changes. Provide instructions so we can reproduce. Please also list any relevant details for your test configuration

• pytest pass.

Test Configuration:

• pytest

Checklist

• [x] pytest done