Searched defs:root_instruction (Results 1 – 3 of 3) sorted by relevance
153 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()
1405 auto root_instruction = function->root_instruction(); in MatchReductionGenerator() local