diff options
author | Adrian Kuegel <akuegel@google.com> | 2018-06-29 07:26:59 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-29 07:29:56 -0700 |
commit | ce979f7916500c0b7c91ed1eb2b118aa0b3f3284 (patch) | |
tree | 72b0f5cc89723520fb55a5f3e8516b472e1d8175 /tensorflow/compiler/xla/tests/BUILD | |
parent | e0af631fc04187543fb4fbe3133069438046b2e2 (diff) |
Handle nested tuples in GpuTransferManager.
This became necessary when the TOKEN primitive type was added.
In some models, an existing tuple T is extended to (T, token[]).
Also add the TOKEN case to a switch statement where it was missing.
PiperOrigin-RevId: 202643759
Diffstat (limited to 'tensorflow/compiler/xla/tests/BUILD')
-rw-r--r-- | tensorflow/compiler/xla/tests/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/tests/BUILD b/tensorflow/compiler/xla/tests/BUILD index 5a45e2e610..20b2885e90 100644 --- a/tensorflow/compiler/xla/tests/BUILD +++ b/tensorflow/compiler/xla/tests/BUILD @@ -2040,6 +2040,7 @@ xla_test( "//tensorflow/compiler/xla:shape_util", "//tensorflow/compiler/xla/client/xla_client:xla_builder", "//tensorflow/compiler/xla/client/xla_client:xla_computation", + "//tensorflow/compiler/xla/service:hlo_parser", "//tensorflow/compiler/xla/tests:xla_internal_test_main", "//tensorflow/core:test", ], |