diff options
author | 2018-06-29 07:26:59 -0700 | |
---|---|---|
committer | 2018-06-29 07:29:56 -0700 | |
commit | ce979f7916500c0b7c91ed1eb2b118aa0b3f3284 (patch) | |
tree | 72b0f5cc89723520fb55a5f3e8516b472e1d8175 /tensorflow/compiler/xla/tests/test_utils.cc | |
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/test_utils.cc')
-rw-r--r-- | tensorflow/compiler/xla/tests/test_utils.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/tests/test_utils.cc b/tensorflow/compiler/xla/tests/test_utils.cc index 000535a982..20c7c30878 100644 --- a/tensorflow/compiler/xla/tests/test_utils.cc +++ b/tensorflow/compiler/xla/tests/test_utils.cc @@ -161,6 +161,9 @@ StatusOr<std::unique_ptr<Literal>> MakeFakeLiteralInternal( })); break; } + // Token requires no data. + case TOKEN: + break; default: return Unimplemented("Unsupported type for fake literal generation: %s", ShapeUtil::HumanString(shape).c_str()); |