message: Load TF specific py_test rule in xla/python/BUILD
diff: load("//tensorflow/core:platform/default/build_config.bzl", "pyx_library")
  load("//tensorflow/compiler/xla:xla.bzl", "xla_py_test_deps", "xla_python_default_plugins")
  load("//tensorflow:tensorflow.bzl", "tf_pybind_extension")
  -load("//tensorflow:tensorflow.bzl", "tf_cc_test")
  +load("//tensorflow:tensorflow.bzl", "py_test", "tf_cc_test")
  package(
  default_visibility = ["//tensorflow:internal"],
  