test_tf2

tf2 unit tests