diff options
| author |  Benjamin Kramer <kramerb@google.com> | 2018-09-20 03:14:04 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-20 03:16:50 -0700 |
| commit | 90d084e0c42232043c186e66093b67800fb30fba (patch) | |
| tree | 230d7998ea42af3efd59b0d25312eaa54efce5de /tensorflow/compiler/tests/BUILD | |
| parent | 9604413da7a27f5718bb88d407d13476dbef5b82 (diff) | |
[XLA:TF] Whitelist quantized types for CPU/GPU
These have the same behavior as unquantized types so we can just pass them
through to XLA (which converts them to unquantized types). They're supposed to
be used with special ops, none of which are currently implemented by XLA.
Casting (without quantization) and basic math works fine though.
These do not have a corresponding numpy type, so only tests using TF types will
see them.
PiperOrigin-RevId: 213781650
Diffstat (limited to 'tensorflow/compiler/tests/BUILD')
| -rw-r--r-- | tensorflow/compiler/tests/BUILD | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD index 97ed554171..3cf74fa788 100644 --- a/tensorflow/compiler/tests/BUILD +++ b/tensorflow/compiler/tests/BUILD @@ -978,7 +978,7 @@ tf_xla_py_test( name = "gather_test", size = "medium", srcs = ["gather_test.py"], - tags = ["noasan"], # times out, http://b/78599043 + tags = ["optonly"], deps = [ ":xla_test", "//tensorflow/python:array_ops", @@ -1198,6 +1198,19 @@ tf_xla_py_test( ) tf_xla_py_test( + name = "quantized_ops_test", + size = "small", + srcs = ["quantized_ops_test.py"], + deps = [ + ":xla_test", + "//tensorflow/python:array_ops", + "//tensorflow/python:dtypes", + "//tensorflow/python:math_ops", + "//tensorflow/python:platform_test", + ], +) + +tf_xla_py_test( name = "xla_ops_test", size = "medium", srcs = ["xla_ops_test.py"], |