diff options
author | Sanjoy Das <sanjoy@google.com> | 2018-08-21 08:52:41 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-21 08:59:26 -0700 |
commit | 9f59beb67643953d87e7673fa0000cc775562693 (patch) | |
tree | 3f30145de2fd5b7969bfc157d76e9b9ca287c9b2 /tensorflow/compiler/tests/BUILD | |
parent | 5b456c9ab567c7d9262c57f32693ff33a87946e6 (diff) |
Allow DT_INT64 input shapes for ReshapeOp.
The fix itself is simple but I'm not sure if this how we should be testing
auto-jit.
PiperOrigin-RevId: 209602427
Diffstat (limited to 'tensorflow/compiler/tests/BUILD')
-rw-r--r-- | tensorflow/compiler/tests/BUILD | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD index ae98b3f0f9..47311d2630 100644 --- a/tensorflow/compiler/tests/BUILD +++ b/tensorflow/compiler/tests/BUILD @@ -388,6 +388,19 @@ tf_xla_py_test( ) tf_xla_py_test( + name = "reshape_op_test", + size = "small", + srcs = ["reshape_op_test.py"], + deps = [ + "//tensorflow/compiler/tests:xla_test", + "//tensorflow/compiler/tf2xla/python:xla", + "//tensorflow/python:array_ops", + "//tensorflow/python:dtypes", + "@absl_py//absl/testing:parameterized", + ], +) + +tf_xla_py_test( name = "dynamic_stitch_test", size = "small", srcs = ["dynamic_stitch_test.py"], |