Searched defs:root_instruction (Results 1 – 6 of 6) sorted by relevance
125 const HloInstruction* root_instruction = in CodegenNestedComputation() local
190 HloInstruction* root_instruction() const { return root_instruction_; } in root_instruction() function
52 HloInstruction* root_instruction) { in Build()70 HloInstruction* root_instruction, HloInstruction* fusion_instruction) in HloComputation()
1753 HloInstruction* root_instruction = entry_computation->root_instruction(); in AddInputAndOutputRequiredAssignments() local1787 const HloInstruction* root_instruction = in AreIntervalsReservedInAlternateMemory() local
1068 auto root_instruction = computation->root_instruction(); in IdentityValueAndHloOpcodeForScatterReduceComputation() local
1370 auto root_instruction = function->root_instruction(); in MatchReductionGenerator() local