[XLA] Update TODO references related to all-reduce.
PiperOrigin-RevId: 216600146
diff --git a/tensorflow/compiler/xla/client/xla_builder.h b/tensorflow/compiler/xla/client/xla_builder.h
index 933c0e7..2916ac1 100644
--- a/tensorflow/compiler/xla/client/xla_builder.h
+++ b/tensorflow/compiler/xla/client/xla_builder.h
@@ -699,7 +699,7 @@
// the same channel_id, they will be 'Allreduce'd. If empty, Allreduce will
// not be applied cross modules.
//
- // TODO(b/79737069): Rename this to AllReduce when it's ready to use.
+ // TODO(b/117564385): Rename this to AllReduce when it's ready to use.
XlaOp CrossReplicaSum(
const XlaOp& operand, const XlaComputation& computation,
absl::Span<const ReplicaGroup> replica_groups = {},
@@ -1864,7 +1864,7 @@
// same channel_id, they will be 'Allreduce'd. If empty, Allreduce will not be
// applied cross modules.
//
-// TODO(b/79737069): Rename this to AllReduce when it's ready to use.
+// TODO(b/117564385): Rename this to AllReduce when it's ready to use.
XlaOp CrossReplicaSum(
const XlaOp& operand, const XlaComputation& computation,
absl::Span<const ReplicaGroup> replica_groups = {},
diff --git a/tensorflow/compiler/xla/service/hlo_instruction.h b/tensorflow/compiler/xla/service/hlo_instruction.h
index 81fe1d0..15a4da8 100644
--- a/tensorflow/compiler/xla/service/hlo_instruction.h
+++ b/tensorflow/compiler/xla/service/hlo_instruction.h
@@ -463,7 +463,7 @@
// the same all_reduce_id, they will be 'Allreduce'd. If empty, Allreduce will
// not be applied cross modules.
//
- // TODO(b/79737069): Rename this to AllReduce.
+ // TODO(b/117564385): Rename this to AllReduce.
static std::unique_ptr<HloInstruction> CreateCrossReplicaSum(
const Shape& shape, absl::Span<HloInstruction* const> operands,
HloComputation* reduce_computation,