message: Disable test on GPU due to segfault
diff: @@ -443,7 +443,10 @@ cuda_py_test(
  srcs = ["function_test.py"],
  python_version = "PY3",
  shard_count = 15,
  - tags = ["nomac"], # b/157056289
  + tags = [
  + "nogpu", # TODO(b/162544929): segfault
  + "nomac", # b/157056289
  + ],
  deps = [
  ":backprop",
  ":cancellation",
  