Searched defs:subcomp (Results 1 – 6 of 6) sorted by relevance
600 bool HloDotDumper::ShouldShowSubcomputation(const HloComputation* subcomp) { in ShouldShowSubcomputation()622 const HloComputation* subcomp, const HloInstruction* parent_instr) { in DumpSubcomputation()712 for (const HloComputation* subcomp : instr->called_computations()) { in DumpComputation() local
144 for (auto* subcomp : instruction->called_computations()) { in RecursivelyRemoveDeadComputations() local
120 for (const HloComputation* subcomp : instr->called_computations()) { in ContainsInstrWithOpcode() local
2081 const HloComputation* subcomp = fused_instructions_computation(); in fused_instructions() local
349 HloComputation* subcomp = in TEST_F() local
747 HloComputation* subcomp = in TEST_F() local