Searched refs:fused_param (Results 1 – 3 of 3) sorted by relevance
684 user->fused_parameters(), [&](HloInstruction* fused_param) { in DoesNotUseOperandBuffer() argument685 return user->operand(fused_param->parameter_number()) == operand; in DoesNotUseOperandBuffer()751 absl::c_find_if(fused_params, [&](HloInstruction* fused_param) { in HasUniqueFusedUseOfOperandAt() argument752 return fusion->operand(fused_param->parameter_number()) == operand; in HasUniqueFusedUseOfOperandAt()757 auto* fused_param = *fused_param_it; in HasUniqueFusedUseOfOperandAt() local760 GetAllUsesOfInstructionAtIndex(fused_param, operand_index); in HasUniqueFusedUseOfOperandAt()
770 for (HloInstruction* fused_param : fused_parameters) { in HandleFusion()771 int64 param_no = fused_param->parameter_number(); in HandleFusion()772 if (!ShapesSame(fused_param->shape(), fusion->operand(param_no)->shape())) { in HandleFusion()1692 for (auto fused_param : fused_parameters) { in CheckFusionInstruction() local1693 int64 param_no = fused_param->parameter_number(); in CheckFusionInstruction()
1723 HloInstruction* fused_param = nullptr; in CloneAndFuseInternal() local1726 fused_param = fused_parameters[i]; in CloneAndFuseInternal()1731 if (fused_param == nullptr) { in CloneAndFuseInternal()1735 fused_param = AddFusionOperand(operand); in CloneAndFuseInternal()1737 TF_CHECK_OK(clone->ReplaceOperandWith(operand_num, fused_param)); in CloneAndFuseInternal()