Searched refs:subcomp (Results 1 – 5 of 5) sorted by relevance
338 bool ShouldShowSubcomputation(const HloComputation* subcomp);345 string DumpSubcomputation(const HloComputation* subcomp,560 bool HloDotDumper::ShouldShowSubcomputation(const HloComputation* subcomp) { in ShouldShowSubcomputation() argument561 if (subcomp->IsFusionComputation()) { in ShouldShowSubcomputation()562 const HloInstruction* fusion = subcomp->FusionInstruction(); in ShouldShowSubcomputation()570 if (!subcomp->IsFusionComputation() && MatchTrivialComputation(subcomp)) { in ShouldShowSubcomputation()576 subcomp->instructions(), in ShouldShowSubcomputation()580 string HloDotDumper::DumpSubcomputation(const HloComputation* subcomp, in DumpSubcomputation() argument582 VLOG(2) << "Dumping subcomputation " << subcomp->name(); in DumpSubcomputation()587 const HloInstruction* from = GetNodeForEdge(subcomp->root_instruction()); in DumpSubcomputation()[all …]
75 for (auto* subcomp : instruction->called_computations()) { in Run() local76 live_computations.insert(subcomp); in Run()
113 for (const HloComputation* subcomp : instr->called_computations()) { in ContainsInstrWithOpcode() local114 if (ContainsInstrWithOpcode(subcomp, opcodes)) { in ContainsInstrWithOpcode()
1354 const HloComputation* subcomp = fused_instructions_computation(); in fused_instructions() local1355 return subcomp->instructions(); in fused_instructions()
602 HloComputation* subcomp = in TEST_F() local605 shape, HloInstruction::FusionKind::kLoop, {x, y}, subcomp); in TEST_F()