diff options
| author |  Benjamin Kramer <kramerb@google.com> | 2018-08-21 12:35:33 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-21 12:43:09 -0700 |
| commit | 4f41091f88cca9c87a627864ccd6962e7bb44313 (patch) | |
| tree | 1b7f9682cac04fc1eb6fe20a37ac52d75cf898df /tensorflow/compiler/xla/service/BUILD | |
| parent | c61a49ec318a42e5740efe566936957126dc04d0 (diff) | |
[XLA] Propagate invalid shape errors through reduce folding and turn it on
HloEvaluator should be stable enough for reduce folding, but it shouldn't
crash when it encounters an instruction without a layout. Verify the layout on
every instruction that gets evaluated and return an error on failure.
PiperOrigin-RevId: 209641401
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 7fdffe85c0..73964733e8 100644 --- a/tensorflow/compiler/xla/service/BUILD +++ b/tensorflow/compiler/xla/service/BUILD @@ -2456,6 +2456,7 @@ tf_cc_test( ":hlo", ":hlo_constant_folding", ":hlo_matchers", + ":hlo_parser", ":hlo_pass", "//tensorflow/compiler/xla:literal", "//tensorflow/compiler/xla:shape_util", |