Searched refs:FusionInstruction (Results 1 – 8 of 8) sorted by relevance
80 instruction->parent()->FusionInstruction(); in Propagate()
469 computation_->FusionInstruction()->name(), ")"); in Header()568 const HloInstruction* fusion = subcomp->FusionInstruction(); in ShouldShowSubcomputation()731 const HloInstruction* fusion = instr->parent()->FusionInstruction(); in TryGetFusionParameterConstant()1250 const HloInstruction* fusion = instr->parent()->FusionInstruction(); in AddInstructionIncomingEdges()
464 HloInstruction* FusionInstruction() const { return fusion_instruction_; } in FusionInstruction() function
158 TEST_F(LayoutAssignmentTest, FusionInstruction) { in TEST_F() argument
1619 if (fusion != fused_computation->FusionInstruction()) { in CheckFusionInstruction()
178 c = c->parent()->FusionInstruction(); in BuffersInvariantWithinConsumer()
386 unnested_hlo = computation->FusionInstruction(); in GetIndexTypeForKernel()
289 std::cout << " Fusion instruction: " << comp->FusionInstruction()->name() in DoInfoCommand()