diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-16 14:13:52 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-16 14:16:53 -0700 |
| commit | bc410d9c0133673e7b93a49487d7e14758cba280 (patch) | |
| tree | 4429ab02626a222b80aab6ab1b426abb70c46e23 /tensorflow/contrib/seq2seq/BUILD | |
| parent | 3d4cddf87d544f4f5868497caf5c6ab3e25aea2b (diff) | |
Use fixed sized tensor arrays and max loop iterations in dynamic_decode if the user supplies it and if the inputs were created in an XLA context.
PiperOrigin-RevId: 193097293
Diffstat (limited to 'tensorflow/contrib/seq2seq/BUILD')
| -rw-r--r-- | tensorflow/contrib/seq2seq/BUILD | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tensorflow/contrib/seq2seq/BUILD b/tensorflow/contrib/seq2seq/BUILD index a62069a252..1a1591d798 100644 --- a/tensorflow/contrib/seq2seq/BUILD +++ b/tensorflow/contrib/seq2seq/BUILD @@ -3,9 +3,12 @@ licenses(["notice"]) # Apache 2.0 -exports_files(["LICENSE"]) +package(default_visibility = [ + "//learning/brain/google/xla/tests:__subpackages__", + "//tensorflow:__subpackages__", +]) -package(default_visibility = ["//tensorflow:__subpackages__"]) +exports_files(["LICENSE"]) load("//tensorflow:tensorflow.bzl", "cuda_py_test") load("//tensorflow:tensorflow.bzl", "tf_custom_op_py_library") @@ -38,6 +41,7 @@ tf_custom_op_py_library( "//tensorflow/python:check_ops", "//tensorflow/python:clip_ops", "//tensorflow/python:control_flow_ops", + "//tensorflow/python:control_flow_util", "//tensorflow/python:embedding_ops", "//tensorflow/python:framework_for_generated_wrappers", "//tensorflow/python:functional_ops", |