Searched refs:subcomp (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_graph_dumper.cc | 344 bool ShouldShowSubcomputation(const HloComputation* subcomp);
351 string DumpSubcomputation(const HloComputation* subcomp,
566 bool HloDotDumper::ShouldShowSubcomputation(const HloComputation* subcomp) { in ShouldShowSubcomputation() argument
567 if (subcomp->IsFusionComputation()) { in ShouldShowSubcomputation()
568 const HloInstruction* fusion = subcomp->FusionInstruction(); in ShouldShowSubcomputation()
577 if (!subcomp->IsFusionComputation() && MatchTrivialComputation(subcomp)) { in ShouldShowSubcomputation()
583 subcomp->instructions(), in ShouldShowSubcomputation()
587 string HloDotDumper::DumpSubcomputation(const HloComputation* subcomp, in DumpSubcomputation() argument
589 VLOG(2) << "Dumping subcomputation " << subcomp->name(); in DumpSubcomputation()
594 const HloInstruction* from = GetNodeForEdge(subcomp->root_instruction()); in DumpSubcomputation()
[all …]
|
| D | hlo_dce.cc | 98 for (auto* subcomp : instruction->called_computations()) { in Run() local
99 live_computations.insert(subcomp); in Run()
|
| D | hlo_query.cc | 115 for (const HloComputation* subcomp : instr->called_computations()) { in ContainsInstrWithOpcode() local
116 if (ContainsInstrWithOpcode(subcomp, opcodes)) { in ContainsInstrWithOpcode()
|
| D | hlo_instructions.cc | 1623 const HloComputation* subcomp = fused_instructions_computation(); in fused_instructions() local
1624 return subcomp->instructions(); in fused_instructions()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fusion_merger_test.cc | 349 HloComputation* subcomp = in TEST_F() local
352 shape, HloInstruction::FusionKind::kLoop, params, subcomp); in TEST_F()
|
| D | multi_output_fusion_test.cc | 747 HloComputation* subcomp = in TEST_F() local
750 shape, HloInstruction::FusionKind::kLoop, {x, y}, subcomp); in TEST_F()
|