Searched refs:fused_param (Results 1 – 3 of 3) sorted by relevance
605 user->fused_parameters(), [&](HloInstruction* fused_param) { in DoesNotUseOperandBuffer() argument606 return user->operand(fused_param->parameter_number()) == operand; in DoesNotUseOperandBuffer()672 absl::c_find_if(fused_params, [&](HloInstruction* fused_param) { in HasUniqueFusedUseOfOperandAt() argument673 return fusion->operand(fused_param->parameter_number()) == operand; in HasUniqueFusedUseOfOperandAt()678 auto* fused_param = *fused_param_it; in HasUniqueFusedUseOfOperandAt() local681 GetAllUsesOfInstructionAtIndex(fused_param, operand_index); in HasUniqueFusedUseOfOperandAt()
522 for (HloInstruction* fused_param : fused_parameters) { in HandleFusion()523 int64 param_no = fused_param->parameter_number(); in HandleFusion()524 if (!ShapesSame(fused_param->shape(), fusion->operand(param_no)->shape())) { in HandleFusion()1255 for (auto fused_param : fused_parameters) { in CheckFusionInstruction() local1256 int64 param_no = fused_param->parameter_number(); in CheckFusionInstruction()
1454 HloInstruction* fused_param = nullptr; in CloneAndFuseInternal() local1457 fused_param = fused_parameters[i]; in CloneAndFuseInternal()1462 if (fused_param == nullptr) { in CloneAndFuseInternal()1466 fused_param = AddFusionOperand(operand); in CloneAndFuseInternal()1468 TF_CHECK_OK(clone->ReplaceOperandWith(operand_num, fused_param)); in CloneAndFuseInternal()