Searched refs:new_root_instruction (Results 1 – 2 of 2) sorted by relevance
377 void HloComputation::set_root_instruction(HloInstruction* new_root_instruction, in set_root_instruction() argument382 CHECK(ShapeUtil::Compatible(new_root_instruction->shape(), in set_root_instruction()384 << new_root_instruction->shape() << " is incompatible with " in set_root_instruction()389 if (new_root_instruction == instruction.get()) { in set_root_instruction()398 if (!Shape::Equal().IgnoreLayout()(new_root_instruction->shape(), in set_root_instruction()402 HloInputOutputAliasConfig(new_root_instruction->shape()); in set_root_instruction()406 root_instruction_ = new_root_instruction; in set_root_instruction()
212 void set_root_instruction(HloInstruction* new_root_instruction,