diff options
author | Mingsheng Hong <hongm@google.com> | 2018-02-07 12:22:55 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-07 12:26:52 -0800 |
commit | 65f676426d808cf20abd6a4a30ad48668ee5fdf2 (patch) | |
tree | 84707b8b38781cb4703e46656f24737d86ac1d2d /tensorflow/c/BUILD | |
parent | 90ce80131a8b5213d9f3eb9649d63921db7874a4 (diff) |
Initial XLA support for TF eager. This is prerequisite for TF compiler's XLA support.
This CL adds XLA support for the following TFE_Op's:
1. A TF op such as MatMul, with full support of constant and resource params.
2. A TF_Function as TFE_Op, where the function must have no constant and
resource params. Removing this restriction requires more discussion and will be
deferred to a later time.
PiperOrigin-RevId: 184877345
Diffstat (limited to 'tensorflow/c/BUILD')
-rw-r--r-- | tensorflow/c/BUILD | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD index c46cb32aa4..314cbc657c 100644 --- a/tensorflow/c/BUILD +++ b/tensorflow/c/BUILD @@ -135,6 +135,10 @@ tf_cuda_library( testonly = 1, srcs = ["c_test_util.cc"], hdrs = ["c_test_util.h"], + visibility = [ + "//learning/brain:__subpackages__", + "//tensorflow:__subpackages__", + ], deps = [ ":c_api", "//tensorflow/core:lib", |