commit | c6128f878f5c7937a00e12ec0f47696fa8b958b0 | [log] [tgz] |
---|---|---|
author | A. Unique TensorFlower <gardener@tensorflow.org> | Sun May 23 17:16:56 2021 -0700 |
committer | TensorFlower Gardener <gardener@tensorflow.org> | Sun May 23 17:22:49 2021 -0700 |
tree | 02fed1b4c7b76b16a85b35620d4b58078d41fd35 | |
parent | b967b1d3468c5251c4a2159aa2ef0428c1dffff3 [diff] |
Remove disabling space-to-batch if graphs have reshapes PiperOrigin-RevId: 375391805 Change-Id: I727a6f8b384cc20902dc1866ea75af87453acb7b
diff --git a/tensorflow/compiler/xla/service/space_to_batch_converter.cc b/tensorflow/compiler/xla/service/space_to_batch_converter.cc index 6e6d8aa..f4e5b7f 100644 --- a/tensorflow/compiler/xla/service/space_to_batch_converter.cc +++ b/tensorflow/compiler/xla/service/space_to_batch_converter.cc
@@ -1345,7 +1345,6 @@ // We can add more non-propagatable opcodes as needed. switch (consumer->opcode()) { case HloOpcode::kCustomCall: - case HloOpcode::kReshape: return true; default: return false;