diff options
author | Benjamin Kramer <kramerb@google.com> | 2018-10-09 15:47:56 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-09 15:51:30 -0700 |
commit | 69c4a426fc4a3afd83c8190467b07c17b8b2ed60 (patch) | |
tree | 1c81b33b71efc63bad8519a77026ac96b805be9e /tensorflow/compiler/xla/service/tuple_points_to_analysis.cc | |
parent | 771955e2b8be98a0b38fada41bd67f663397c87d (diff) |
[XLA] Allow scatter to share the operand buffer with the output
This avoids a copy.
PiperOrigin-RevId: 216437329
Diffstat (limited to 'tensorflow/compiler/xla/service/tuple_points_to_analysis.cc')
-rw-r--r-- | tensorflow/compiler/xla/service/tuple_points_to_analysis.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/tuple_points_to_analysis.cc b/tensorflow/compiler/xla/service/tuple_points_to_analysis.cc index 811ac55e2d..ef4e69180d 100644 --- a/tensorflow/compiler/xla/service/tuple_points_to_analysis.cc +++ b/tensorflow/compiler/xla/service/tuple_points_to_analysis.cc @@ -756,6 +756,7 @@ bool TuplePointsToAnalysis::CanShareOperandBufferWithUser( } } if (user->opcode() == HloOpcode::kDynamicUpdateSlice || + user->opcode() == HloOpcode::kScatter || user->opcode() == HloOpcode::kWhile) { // We eliminated other users in BufferLiveness::live_range_strictly_before, // so here we just need to check that the use is at operand index 0. |