Searched refs:root_inst (Results 1 – 2 of 2) sorted by relevance
1219 HloInstruction* root_inst = branch_computation->root_instruction(); in BoundariesToMoveInOrOut() local1220 CHECK(root_inst != nullptr); in BoundariesToMoveInOrOut()1221 boundary_in.mutable_operands().push_back(root_inst); in BoundariesToMoveInOrOut()
3312 const HloInstruction* root_inst = callee.root_instruction(); in GetBufferForGlobalCallReturnValue() local3313 if (root_inst->opcode() == HloOpcode::kOutfeed) { in GetBufferForGlobalCallReturnValue()3318 assignment_.GetUniqueTopLevelSlice(root_inst).ValueOrDie(); in GetBufferForGlobalCallReturnValue()3319 return EmitBufferPointer(root_buffer, root_inst->shape()); in GetBufferForGlobalCallReturnValue()