Rollback copy insertion change because it results in a DCHECK with an internal model.

END_PUBLIC

BEGIN_PUBLIC
Automated g4 rollback of changelist 174423881

PiperOrigin-RevId: 174505237

1 files changed, 2 insertions, 8 deletions

diff --git a/tensorflow/compiler/xla/service/hlo_computation.h b/tensorflow/compiler/xla/service/hlo_computation.h
index 3208197f89..c9782cc981 100644
--- a/tensorflow/compiler/xla/service/hlo_computation.h
+++ b/tensorflow/compiler/xla/service/hlo_computation.h
@@ -152,18 +152,12 @@ class HloComputation {
   //   computation_map: a map from computation name to HloComputation*. This map
   //     must contain all computations which the newly constructed computation
   //     calls.
-  //   add_fused_computation: A function to call to add a fused
-  //     computation. Used (clearly) when the instruction is a fusion
-  //     instruction.
-  //   fusion_instruction: if non-null then the newly created computation will
-  //   be
+  //  fusion_instruction: if non-null then the newly created computation will be
   //     constructed as a fused computation with this instruction as its fusion
   //     parent.
   static StatusOr<std::unique_ptr<HloComputation>> CreateFromProto(
       HloModule* module, const HloComputationProto& proto,
-      const tensorflow::gtl::FlatMap<string, HloComputation*>& computation_map,
-      const std::function<void(std::unique_ptr<HloComputation>)>&
-          add_fused_computation,
+      tensorflow::gtl::FlatMap<string, HloComputation*>* computation_map,
       HloInstruction* fusion_instruction = nullptr);
 
   // Gets the instructions in this computation.