Searched defs:root_instruction (Results 1 – 5 of 5) sorted by relevance
117 const HloInstruction* root_instruction = in CodegenNestedComputation() local
189 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()
557 HloInstruction* root_instruction = entry_computation->root_instruction(); in AddInputAndOutputRequiredAssignments() local591 const HloInstruction* root_instruction = in AreIntervalsReservedInAlternateMemory() local
1593 auto root_instruction = function->root_instruction(); in MatchReductionGenerator() local