Searched refs:fused_instructions (Results 1 – 11 of 11) sorted by relevance
90 for (auto* fused_instruction : fusion->fused_instructions()) { in CalculateBytesReadByFusionInstruction()226 if (absl::c_any_of(fusion->fused_instructions(), in HandleFusion()
34 for (auto* fused : instruction->fused_instructions()) { in GatherFusionInstructions()
127 for (auto* fused : instruction->fused_instructions()) { in GatherFusionInstructions()169 PopulateDefinedBuffersAndAliases(instruction->fused_instructions())); in Analyze()557 for (auto* fused : instruction->fused_instructions()) { in ToString()
215 for (const auto& fused_instruction : instruction->fused_instructions()) { in PerformSinkReshapeOrTranspose()
1181 for (auto* fused : fused_instructions()) { in IsElementwiseImpl()1243 auto fused_instructions = cloned_fusion->fused_instructions_computation() in MergeFusionInstruction() local1245 for (auto fused_it = fused_instructions.rbegin(); in MergeFusionInstruction()1246 fused_it != fused_instructions.rend(); ++fused_it) { in MergeFusionInstruction()1353 HloFusionInstruction::fused_instructions() const { in fused_instructions() function in xla::HloFusionInstruction1360 HloFusionInstruction::fused_instructions() { in fused_instructions() function in xla::HloFusionInstruction
1460 fused_instructions() const;1464 fused_instructions();
726 fusion->fused_instructions(), [&](const HloInstruction* fused) { in GetFusionParameterForOperand()730 CHECK(it != fusion->fused_instructions().end()); in GetFusionParameterForOperand()
767 fused_instructions() const;771 fused_instructions();
3328 HloInstruction::fused_instructions() const { in fused_instructions() function in xla::HloInstruction3329 return Cast<HloFusionInstruction>(this)->fused_instructions(); in fused_instructions()3334 HloInstruction::fused_instructions() { in fused_instructions() function in xla::HloInstruction3335 return Cast<HloFusionInstruction>(this)->fused_instructions(); in fused_instructions()
1292 for (const auto& fused : fusion->fused_instructions()) { in CheckFusionInstruction()
276 std::transform(root->fused_instructions().begin(), in RunFusionAndCheckOpcodesWereFused()277 root->fused_instructions().end(), fused_opcodes.begin(), in RunFusionAndCheckOpcodesWereFused()