diff options
author | 2017-12-01 19:38:36 -0800 | |
---|---|---|
committer | 2017-12-01 19:42:18 -0800 | |
commit | 01f097d789e88c58cfc16d5052e2bb83f6412ef3 (patch) | |
tree | 16e6da24a7e20d551fbd18b61bd3cc7efe1a376d | |
parent | 5edb6fe8bcce4a9070baef862fba097b807ffbd8 (diff) |
Automated g4 rollback of changelist 177204171
PiperOrigin-RevId: 177670554
-rw-r--r-- | tensorflow/compiler/xla/service/hlo_verifier.cc | 10 |
1 files changed, 3 insertions, 7 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_verifier.cc b/tensorflow/compiler/xla/service/hlo_verifier.cc index ea7775b18a..4d02846cf7 100644 --- a/tensorflow/compiler/xla/service/hlo_verifier.cc +++ b/tensorflow/compiler/xla/service/hlo_verifier.cc @@ -147,13 +147,9 @@ class ShapeVerifier : public DfsHloVisitor { } Status HandleBitcast(HloInstruction* bitcast) override { - // Bitcasts that are not the root of a computation can be any shape. - // Bitcasts that are the root of a computation must have the same shape - // byte size as their operand. - if (bitcast->parent()->root_instruction() == bitcast) { - TF_RET_CHECK(shape_size_fn_(bitcast->shape()) == - shape_size_fn_(bitcast->operand(0)->shape())); - } + // Bitcasts can be any shape, as long as the size matches the operand size. + TF_RET_CHECK(shape_size_fn_(bitcast->shape()) == + shape_size_fn_(bitcast->operand(0)->shape())); return tensorflow::Status::OK(); } |