Searched refs:ReplaceUseWithDifferentShape (Results 1 – 6 of 6) sorted by relevance
54 operand->ReplaceUseWithDifferentShape(instruction, domain)); in RunInternal()
161 TF_RETURN_IF_ERROR(hlo->ReplaceUseWithDifferentShape(user, new_hlo)); in InsertConvertAfterOutput()208 TF_RETURN_IF_ERROR(hlo->ReplaceUseWithDifferentShape(user, new_hlo)); in ChangeOutputTypeThenInsertConvertBack()
941 TF_RETURN_IF_ERROR(inst->ReplaceUseWithDifferentShape(user, copy)); in Run()
1189 Status ReplaceUseWithDifferentShape(HloInstruction* user,
2307 return ReplaceUseWithDifferentShape(user, new_producer); in ReplaceUseWith()2310 Status HloInstruction::ReplaceUseWithDifferentShape( in ReplaceUseWithDifferentShape() function in xla::HloInstruction2371 TF_RETURN_IF_ERROR(ReplaceUseWithDifferentShape(user, new_producer)); in ReplaceAllUsesWithDifferentShape()
887 TF_RETURN_IF_ERROR(param->ReplaceUseWithDifferentShape(user, param_copy)); in CopyOperandIfLayoutsDiffer()