Searched refs:fused_parameter (Results 1 – 13 of 13) sorted by relevance
173 auto fused_parameter = consumer->fused_parameter(producer_operand_index); in ShouldFuseInexpensiveChecks() local175 fused_parameter->users(); in ShouldFuseInexpensiveChecks()
3352 const HloInstruction* input = fusion->fused_parameter(input_ids[i]); in FilterInputsForShmemTranspose()
137 auto* fused_parameter = inst->fused_parameter(i); in RevertIfFusionInternalBF16Changes() local138 if (has_changes(fused_parameter)) { in RevertIfFusionInternalBF16Changes()139 changes_to_bf16_.erase(fused_parameter); in RevertIfFusionInternalBF16Changes()248 auto* fused_parameter = in AllUsersConsumeBF16() local249 use.instruction->fused_parameter(use.operand_number); in AllUsersConsumeBF16()250 if (OutputTypeAfterChange(fused_parameter, use.operand_index) != BF16) { in AllUsersConsumeBF16()
1192 worklist.push_back(fused_parameter(operand_idx.value())); in IsElementwiseImpl()1193 visited.insert(fused_parameter(operand_idx.value())); in IsElementwiseImpl()1223 HloInstruction* fused_parameter = in AddFusionOperand() local1228 return fused_parameter; in AddFusionOperand()1340 HloInstruction* HloFusionInstruction::fused_parameter( in fused_parameter() function in xla::HloFusionInstruction1418 HloInstruction* fused_parameter = fused_parameters[operand_num]; in CloneAndFuseInternal() local1419 TF_CHECK_OK(fused_parameter->ReplaceAllUsesWith(clone)); in CloneAndFuseInternal()1585 TF_RETURN_IF_ERROR(fused_parameter(i)->ReplaceAllUsesWith( in DeduplicateFusionOperands()1586 fused_parameter(emplace_result.first->second))); in DeduplicateFusionOperands()
935 user->fused_parameter(use.operand_number); in DoesNotUseOperandBuffer()1038 user->fused_parameter(user->operand_index(operand)); in CanShareOperandBufferWithUser()
199 layout, fusion->fused_parameter(0)->shape().layout())); in TEST_F()201 layout, fusion->fused_parameter(1)->shape().layout())); in TEST_F()
732 user->fused_parameter(user->operand_index(operand)); in CanShareOperandBufferWithUser()
1470 HloInstruction* fused_parameter(int64 parameter_number) const;
778 HloInstruction* fused_parameter(int64 parameter_number) const;
3342 HloInstruction* HloInstruction::fused_parameter(int64 parameter_number) const { in fused_parameter() function in xla::HloInstruction3343 return Cast<HloFusionInstruction>(this)->fused_parameter(parameter_number); in fused_parameter()
737 fusion->fused_parameter(i)->shape(), fusion->operand(i)->shape())); in CheckFusionLayout()
277 auto instruction = consumer->fused_parameter(operand_num); in IsFusedIrEmitterInefficient()
410 instruction->fused_parameter(op_num); in FindConstrainedUses()