diff options
| author |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-08 12:26:34 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-08 12:26:34 -0700 |
| commit | fb2807c70de107eb55fb6e8fff2bb8a7e4284ea9 (patch) | |
| tree | b1e1af2df117b0d93eefeb24b009c4ab52ec2027 /tensorflow/compiler | |
| parent | dcd3b4307a3095e3f18aef53f5034787e3cc3af6 (diff) | |
| parent | a12b8c4afdca3ac2945d62b3b83ca2599ab360f9 (diff) | |
Merge pull request #22303 from JuliaComputing:kf/broadcastshapeval
PiperOrigin-RevId: 216228494
Diffstat (limited to 'tensorflow/compiler')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_verifier.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_verifier.cc b/tensorflow/compiler/xla/service/hlo_verifier.cc index fad3b14ec2..496fe1795d 100644 --- a/tensorflow/compiler/xla/service/hlo_verifier.cc +++ b/tensorflow/compiler/xla/service/hlo_verifier.cc @@ -313,8 +313,9 @@ Status ShapeVerifier::HandleBroadcast(HloInstruction* broadcast) { operand_dimension < ShapeUtil::Rank(operand_shape); ++operand_dimension) { int64 output_dimension = broadcast->dimensions()[operand_dimension]; - TF_RET_CHECK(broadcast->shape().dimensions(output_dimension) == - operand_shape.dimensions(operand_dimension)) + TF_RET_CHECK((output_dimension < ShapeUtil::Rank(broadcast->shape())) && + (broadcast->shape().dimensions(output_dimension) == + operand_shape.dimensions(operand_dimension))) << broadcast->ToString() << " operand shape " << operand_shape; } return Status::OK(); |