diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-12-01 19:38:36 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-01 19:42:18 -0800 |
| commit | 01f097d789e88c58cfc16d5052e2bb83f6412ef3 (patch) | |
| tree | 16e6da24a7e20d551fbd18b61bd3cc7efe1a376d /tensorflow/compiler/xla/service/hlo_verifier.cc | |
| parent | 5edb6fe8bcce4a9070baef862fba097b807ffbd8 (diff) | |
Automated g4 rollback of changelist 177204171
PiperOrigin-RevId: 177670554
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_verifier.cc')
| -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(); } |