Searched refs:in_place_op (Results 1 – 1 of 1) sorted by relevance
338 HloInstruction* in_place_op, in AddCopiesForInPlaceOperation() argument340 VLOG(2) << "Adding copies for in-place operation " << in_place_op->name(); in AddCopiesForInPlaceOperation()341 HloInstruction* operand = in_place_op->mutable_operand(operand_number); in AddCopiesForInPlaceOperation()343 in_place_op->parent()->DeepCopyInstruction(operand)); in AddCopiesForInPlaceOperation()344 TF_RETURN_IF_ERROR(operand->ReplaceUseWith(in_place_op, deep_copy)); in AddCopiesForInPlaceOperation()