Update BUILD

This commit is contained in:
geetachavan1 2020-11-17 12:22:41 -08:00 committed by GitHub
parent 117698df2a
commit 4041e1788c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -24,10 +24,10 @@ tf_py_test(
name = "resource_ops_test",
size = "small",
srcs = ["resource_ops_test.py"],
tfrt_enabled = True,
tags = [
"no_mac", # b/173427076
],
tfrt_enabled = True,
deps = [
"//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_py",
"//tensorflow/python:boosted_trees_ops",
@ -43,10 +43,10 @@ tf_py_test(
name = "prediction_ops_test",
size = "small",
srcs = ["prediction_ops_test.py"],
tfrt_enabled = True,
tags = [
"no_mac", # b/173427076
],
tfrt_enabled = True,
deps = [
"//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_py",
"//tensorflow/python:array_ops",
@ -78,10 +78,10 @@ tf_py_test(
name = "training_ops_test",
size = "small",
srcs = ["training_ops_test.py"],
tfrt_enabled = True,
tags = [
"no_mac", # b/173427076
],
tfrt_enabled = True,
deps = [
"//tensorflow/core/kernels/boosted_trees:boosted_trees_proto_py",
"//tensorflow/python:array_ops",