diff options
| -rw-r--r-- | tensorflow/BUILD | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/tensorflow/BUILD b/tensorflow/BUILD index 233fe21fbf..0bce474dfa 100644 --- a/tensorflow/BUILD +++ b/tensorflow/BUILD @@ -590,10 +590,16 @@ py_library( cc_library( name = "grpc", - deps = ["@grpc"], + deps = select({ + ":linux_s390x": ["@grpc//:grpc_unsecure"], + "//conditions:default": ["@grpc"], + }), ) cc_library( name = "grpc++", - deps = ["@grpc//:grpc++"], + deps = select({ + ":linux_s390x": ["@grpc//:grpc++_unsecure"], + "//conditions:default": ["@grpc//:grpc++"], + }), ) |