Searched refs:fused_instructions (Results 1 – 12 of 12) sorted by relevance
91 for (auto* fused_instruction : fusion->fused_instructions()) { in CalculateBytesReadByFusionInstruction()254 absl::c_any_of(fusion->fused_instructions(), in HandleFusion()
450 for (const auto& instr : producer.fused_instructions()) { in CreatesNestedLoop()
36 for (auto* fused : instruction->fused_instructions()) { in GatherFusionInstructions()
130 for (auto* fused : instruction->fused_instructions()) { in GatherFusionInstructions()172 PopulateDefinedBuffersAndAliases(instruction->fused_instructions())); in Analyze()637 for (auto* fused : instruction->fused_instructions()) { in ToString()
216 for (const auto& fused_instruction : instruction->fused_instructions()) { in PerformSinkReshapeOrTranspose()
1414 for (auto* fused : fused_instructions()) { in IsElementwiseImpl()1473 auto fused_instructions = cloned_fusion->fused_instructions_computation() in MergeFusionInstruction() local1475 for (auto fused_it = fused_instructions.rbegin(); in MergeFusionInstruction()1476 fused_it != fused_instructions.rend(); ++fused_it) { in MergeFusionInstruction()1622 HloFusionInstruction::fused_instructions() const { in fused_instructions() function in xla::HloFusionInstruction1629 HloFusionInstruction::fused_instructions() { in fused_instructions() function in xla::HloFusionInstruction
777 fusion->fused_instructions(), [&](const HloInstruction* fused) { in GetFusionParameterForOperand()781 CHECK(it != fusion->fused_instructions().end()); in GetFusionParameterForOperand()
1740 fused_instructions() const;1744 fused_instructions();
951 fused_instructions() const;955 fused_instructions();
4023 HloInstruction::fused_instructions() const { in fused_instructions() function in xla::HloInstruction4024 return Cast<HloFusionInstruction>(this)->fused_instructions(); in fused_instructions()4029 HloInstruction::fused_instructions() { in fused_instructions() function in xla::HloInstruction4030 return Cast<HloFusionInstruction>(this)->fused_instructions(); in fused_instructions()
1729 for (const auto& fused : fusion->fused_instructions()) { in CheckFusionInstruction()
278 std::transform(root->fused_instructions().begin(), in RunFusionAndCheckOpcodesWereFused()279 root->fused_instructions().end(), fused_opcodes.begin(), in RunFusionAndCheckOpcodesWereFused()