Fix unit test failures
Fixes test failure introduced with https://github.com/tensorflow/tensorflow/commit/ce26e300f41d831606bf31b9095ef769486c23cf
Add tag to tensor_float_32_test to prevent its running on pip where it will fail due to being unable to import xla_test from tensorflow.compiler.tests