Searched refs:new_root_instruction (Results 1 – 2 of 2) sorted by relevance
275 void HloComputation::set_root_instruction(HloInstruction* new_root_instruction, in set_root_instruction() argument280 CHECK(ShapeUtil::Compatible(new_root_instruction->shape(), in set_root_instruction()282 << new_root_instruction->shape() << " is incompatible with " in set_root_instruction()287 if (new_root_instruction == instruction.get()) { in set_root_instruction()294 root_instruction_ = new_root_instruction; in set_root_instruction()
148 void set_root_instruction(HloInstruction* new_root_instruction,