Searched refs:ReplaceAllUsesWithDifferentShape (Results 1 – 5 of 5) sorted by relevance
352 TF_RETURN_IF_ERROR(convert_or_clamp->ReplaceAllUsesWithDifferentShape(gte)); in RewriteForConvertOrClampImpl()379 TF_RETURN_IF_ERROR(side_converter->ReplaceAllUsesWithDifferentShape( in RewriteForFinalOutput()
1218 Status ReplaceAllUsesWithDifferentShape(HloInstruction* new_producer);1223 Status ReplaceAllUsesWithDifferentShape(
2365 return ReplaceAllUsesWithDifferentShape(users, new_producer); in ReplaceUsesWith()2368 Status HloInstruction::ReplaceAllUsesWithDifferentShape( in ReplaceAllUsesWithDifferentShape() function in xla::HloInstruction2385 return ReplaceAllUsesWithDifferentShape(new_producer); in ReplaceAllUsesWith()2388 Status HloInstruction::ReplaceAllUsesWithDifferentShape( in ReplaceAllUsesWithDifferentShape() function in xla::HloInstruction
496 conditional->ReplaceAllUsesWithDifferentShape(newconditional)); in ConvertSpecialMove()
1774 TF_CHECK_OK(ReplaceAllUsesWithDifferentShape(new_instr)); in CloneAndFuseInternal()