Searched defs:root_instruction (Results 1 – 6 of 6) sorted by relevance
134 const HloInstruction* root_instruction = in CodegenNestedComputation() local
60 HloInstruction* root_instruction) { in Build()78 HloInstruction* root_instruction, HloInstruction* fusion_instruction) in HloComputation()
217 HloInstruction* root_instruction() const { return root_instruction_; } in root_instruction() function
2193 HloInstruction* root_instruction = in AllocateCrossProgramPrefetchBuffer() local2427 HloInstruction* root_instruction = entry_computation->root_instruction(); in AddInputAndOutputRequiredAssignments() local2520 const HloInstruction* root_instruction = in AreIntervalsReservedInAlternateMemory() local
1183 auto root_instruction = computation->root_instruction(); in IdentityValueAndHloOpcodeForScatterReduceComputation() local
1464 auto root_instruction = function->root_instruction(); in MatchReductionGenerator() local