diff options
Diffstat (limited to 'tensorflow/compiler/tests/build_defs.bzl')
-rw-r--r-- | tensorflow/compiler/tests/build_defs.bzl | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/compiler/tests/build_defs.bzl b/tensorflow/compiler/tests/build_defs.bzl index 0bde616521..a56c53de0f 100644 --- a/tensorflow/compiler/tests/build_defs.bzl +++ b/tensorflow/compiler/tests/build_defs.bzl @@ -59,6 +59,9 @@ def tf_xla_py_test(name, srcs=[], deps=[], tags=[], data=[], main=None, backend_args += ["--test_device=" + plugins[backend]["device"], "--types=" + plugins[backend]["types"]] backend_tags += plugins[backend]["tags"] + backend_args += plugins[backend]["args"] + backend_deps += plugins[backend]["deps"] + backend_data += plugins[backend]["data"] else: fail("Unknown backend {}".format(backend)) |