diff options
author | 2018-05-16 19:12:18 -0700 | |
---|---|---|
committer | 2018-05-16 19:14:49 -0700 | |
commit | cf55582ed8ccaa39f70d7370513050bcf65411be (patch) | |
tree | a9487c57b0c4e765baada80a4a8332013118a9c1 /tensorflow/compiler/tests/BUILD | |
parent | b33d1001b3e03c454cf28c5ae8f87ace608d849e (diff) |
Enhance DenseLayer + XLA compatibility test cases to cover compilation behavior differences in different jit modes.
PiperOrigin-RevId: 196926896
Diffstat (limited to 'tensorflow/compiler/tests/BUILD')
-rw-r--r-- | tensorflow/compiler/tests/BUILD | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/tensorflow/compiler/tests/BUILD b/tensorflow/compiler/tests/BUILD index 213ab95a12..2a88743c80 100644 --- a/tensorflow/compiler/tests/BUILD +++ b/tensorflow/compiler/tests/BUILD @@ -853,9 +853,21 @@ cuda_py_test( "//tensorflow/python:control_flow_ops", "//tensorflow/python:framework", "//tensorflow/python:gradients", - "//tensorflow/python:layers", "//tensorflow/python:math_ops", "//tensorflow/python:nn_ops", + ], +) + +cuda_py_test( + name = "dense_layer_test", + size = "small", + srcs = ["dense_layer_test.py"], + additional_deps = [ + "//tensorflow/contrib/compiler:compiler_py", + "//tensorflow/core:protos_all_py", + "//tensorflow/python:array_ops", + "//tensorflow/python:client_testlib", + "//tensorflow/python:layers", "//tensorflow/python:variables", ], ) |