Searched refs:FusionInstruction (Results 1 – 7 of 7) sorted by relevance
463 computation_->FusionInstruction()->name(), ")"); in Header()562 const HloInstruction* fusion = subcomp->FusionInstruction(); in ShouldShowSubcomputation()722 const HloInstruction* fusion = instr->parent()->FusionInstruction(); in TryGetFusionParameterConstant()1211 const HloInstruction* fusion = instr->parent()->FusionInstruction(); in AddInstructionIncomingEdges()
398 HloInstruction* FusionInstruction() const { return fusion_instruction_; } in FusionInstruction() function
1182 if (fusion != fused_computation->FusionInstruction()) { in CheckFusionInstruction()
158 TEST_F(LayoutAssignmentTest, FusionInstruction) { in TEST_F() argument
200 c = c->parent()->FusionInstruction(); in BuffersInvariantWithinConsumer()
299 unnested_hlo = computation->FusionInstruction(); in GetIndexTypeForKernel()1178 scatter->IsFused() ? scatter->parent()->FusionInstruction() : scatter; in EmitScatter()
263 std::cout << " Fusion instruction: " << comp->FusionInstruction()->name() in DoInfoCommand()