diff options
| author |  Bixia Zheng <bixia@google.com> | 2018-10-05 11:49:19 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-05 11:54:17 -0700 |
| commit | 03b4161326897453fa6b2803b873954607f7623b (patch) | |
| tree | 26753b2411e13844541948aa59a293cd47cf7229 /tensorflow/compiler/xla/service/BUILD | |
| parent | 496bc1589831da2f00e6d49b12c68b97301730d4 (diff) | |
[XLA] Extend the HLO verifier to check that non-layout-changing instructions
preserve operand layouts.
Add an std::function member to the HloVerifier for a backend to specify the
function object used to determine whether an instruction can change layouts.
Use the function object to find out the non-layout-changing instructions and
check that such instructions should produce results with the same layouts as
its operands.
Add test cases.
PiperOrigin-RevId: 215941282
Diffstat (limited to 'tensorflow/compiler/xla/service/BUILD')
| -rw-r--r-- | tensorflow/compiler/xla/service/BUILD | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/BUILD b/tensorflow/compiler/xla/service/BUILD index 4797cf3330..2b292ed053 100644 --- a/tensorflow/compiler/xla/service/BUILD +++ b/tensorflow/compiler/xla/service/BUILD @@ -2450,6 +2450,7 @@ tf_cc_test( ":hlo", ":hlo_parser", ":hlo_verifier", + ":layout_assignment", "//tensorflow/compiler/xla:shape_util", "//tensorflow/compiler/xla:test", "//tensorflow/compiler/xla:types", |