Searched refs:fused_parameter (Results 1 – 15 of 15) sorted by relevance
37 instruction->fused_parameter(operand_idx), in Run()103 use.instruction->fused_parameter(use.operand_number), memory_space); in Propagate()
139 auto* fused_parameter = inst->fused_parameter(i); in RevertIfFusionInternalBF16Changes() local140 if (has_changes(fused_parameter)) { in RevertIfFusionInternalBF16Changes()141 changes_to_bf16_.erase(fused_parameter); in RevertIfFusionInternalBF16Changes()277 auto* fused_parameter = in AllUsersConsumeBF16() local278 use.instruction->fused_parameter(use.operand_number); in AllUsersConsumeBF16()279 if (OutputTypeAfterChange(fused_parameter, use.operand_index) != BF16) { in AllUsersConsumeBF16()
73 cnt += user->fused_parameter(operand_index)->user_count(); in UserCount()
1425 worklist.push_back(fused_parameter(operand_idx.value())); in IsElementwiseImpl()1426 visited.insert(fused_parameter(operand_idx.value())); in IsElementwiseImpl()1453 HloInstruction* fused_parameter = in AddFusionOperand() local1458 return fused_parameter; in AddFusionOperand()1609 HloInstruction* HloFusionInstruction::fused_parameter( in fused_parameter() function in xla::HloFusionInstruction1687 HloInstruction* fused_parameter = fused_parameters[operand_num]; in CloneAndFuseInternal() local1688 TF_CHECK_OK(fused_parameter->ReplaceAllUsesWith(clone)); in CloneAndFuseInternal()1851 TF_RETURN_IF_ERROR(fused_parameter(i)->ReplaceAllUsesWith( in DeduplicateFusionOperands()1852 fused_parameter(emplace_result.first->second))); in DeduplicateFusionOperands()
155 size += FusionParameterReadBytes(user->fused_parameter(idx)); in FusionParameterReadBytes()898 const HloInstruction* operand = fusion->fused_parameter(i); in HandleFusion()
1204 user->fused_parameter(use.operand_number); in DoesNotUseOperandBuffer()1302 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()
1750 HloInstruction* fused_parameter(int64 parameter_number) const;
962 HloInstruction* fused_parameter(int64 parameter_number) const;
4037 HloInstruction* HloInstruction::fused_parameter(int64 parameter_number) const { in fused_parameter() function in xla::HloInstruction4038 return Cast<HloFusionInstruction>(this)->fused_parameter(parameter_number); in fused_parameter()
755 TF_RET_CHECK(LayoutsInShapesEqual(fusion->fused_parameter(i)->shape(), in CheckFusionLayout()
49 LooksLikeAnActivation(user->fused_parameter(i))) { in LooksLikeAnActivation()
690 fusion->fused_parameter(i), in BindFusionArguments()
519 instruction->fused_parameter(op_num); in FindConstrainedUses()
3327 fusion->fused_parameter(i), in BindFusionArguments()