Searched refs:fused_parameter (Results 1 – 16 of 16) sorted by relevance
45 instruction->fused_parameter(operand_idx), in Run()111 use.instruction->fused_parameter(use.operand_number), memory_space); in Propagate()
135 auto* fused_parameter = inst->fused_parameter(i); in RevertIfFusionInternalBF16Changes() local136 if (has_changes(fused_parameter)) { in RevertIfFusionInternalBF16Changes()137 changes_to_bf16_.erase(fused_parameter); in RevertIfFusionInternalBF16Changes()273 auto* fused_parameter = in AllUsersConsumeBF16() local274 use.instruction->fused_parameter(use.operand_number); in AllUsersConsumeBF16()275 if (OutputTypeAfterChange(fused_parameter, use.operand_index) != BF16) { in AllUsersConsumeBF16()
73 cnt += user->fused_parameter(operand_index)->user_count(); in UserCount()
155 size += FusionParameterReadBytes(user->fused_parameter(idx)); in FusionParameterReadBytes()956 const HloInstruction* operand = fusion->fused_parameter(i); in HandleFusion()
1633 user->fused_parameter(use.operand_number); in DoesNotUseOperandBuffer()1858 user->fused_parameter(user->operand_index(operand)); in CanShareOperandBufferWithUser()1902 user->fused_parameter(user->operand_index(operand)); in CanShareOperandBufferWithUser()
1892 worklist.push_back(fused_parameter(operand_idx.value())); in IsElementwiseImpl()1893 visited.insert(fused_parameter(operand_idx.value())); in IsElementwiseImpl()2067 HloInstruction* HloFusionInstruction::fused_parameter( in fused_parameter() function in xla::HloFusionInstruction2129 TF_RETURN_IF_ERROR(fused_parameter(i)->ReplaceAllUsesWith( in DeduplicateFusionOperands()2130 fused_parameter(emplace_result.first->second))); in DeduplicateFusionOperands()
200 layout, fusion->fused_parameter(0)->shape().layout())); in TEST_F()202 layout, fusion->fused_parameter(1)->shape().layout())); in TEST_F()
888 has_nonelementwise_uses_except_dus(consumer->fused_parameter(i))) { in IsSafeToFuseSliceIntoDusFusion()
1966 HloInstruction* fused_parameter(int64_t parameter_number) const;
1291 HloInstruction* fused_parameter(int64_t parameter_number) const;
4609 HloInstruction* HloInstruction::fused_parameter( in fused_parameter() function in xla::HloInstruction4611 return Cast<HloFusionInstruction>(this)->fused_parameter(parameter_number); in fused_parameter()
917 TF_RET_CHECK(LayoutsInShapesEqual(fusion->fused_parameter(i)->shape(), in CheckFusionLayout()
65 LooksLikeAnActivation(user->fused_parameter(i))) { in LooksLikeAnActivation()
680 *fusion->fused_parameter(i), in BindFusionArguments()
568 instruction->fused_parameter(op_num); in FindConstrainedUses()
3546 *fusion->fused_parameter(i), in BindFusionArguments()