diff options
author | Asim Shankar <ashankar@google.com> | 2017-01-25 23:25:29 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-01-25 23:47:45 -0800 |
commit | 870b7775b2538d78d9353eba53d7664021405013 (patch) | |
tree | aaf53380e314526af84327f777e276054e7dfdc1 /tensorflow/java/BUILD | |
parent | fb53109ad20b2e568246b05796296a047c271856 (diff) |
Java: Improved Shape support.
- Support fetching shapes of outputs of operations in a graph
- Support setting a shape valued attribute of an operation
Also, bugfix for seting list-valued attributes which would have resulted
in an UnsatisfiedLinkError prior to this commit. This commit fixes the bug
and adds some tests for setting attribute values as well.
Another step towards #5.
Also addresses a follow-up request in #6781
Change: 145640678
Diffstat (limited to 'tensorflow/java/BUILD')
-rw-r--r-- | tensorflow/java/BUILD | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/tensorflow/java/BUILD b/tensorflow/java/BUILD index af3d2aec09..134c173b62 100644 --- a/tensorflow/java/BUILD +++ b/tensorflow/java/BUILD @@ -67,6 +67,18 @@ java_test( ) java_test( + name = "ShapeTest", + size = "small", + srcs = ["src/test/java/org/tensorflow/ShapeTest.java"], + test_class = "org.tensorflow.ShapeTest", + deps = [ + ":tensorflow", + ":testutil", + "//external:junit", + ], +) + +java_test( name = "TensorFlowTest", size = "small", srcs = ["src/test/java/org/tensorflow/TensorFlowTest.java"], |