Searched refs:new_root_instruction (Results 1 – 2 of 2) sorted by relevance
341 void HloComputation::set_root_instruction(HloInstruction* new_root_instruction, in set_root_instruction() argument346 CHECK(ShapeUtil::Compatible(new_root_instruction->shape(), in set_root_instruction()348 << new_root_instruction->shape() << " is incompatible with " in set_root_instruction()353 if (new_root_instruction == instruction.get()) { in set_root_instruction()362 if (!Shape::Equal().IgnoreLayout()(new_root_instruction->shape(), in set_root_instruction()366 HloInputOutputAliasConfig(new_root_instruction->shape()); in set_root_instruction()370 root_instruction_ = new_root_instruction; in set_root_instruction()
191 void set_root_instruction(HloInstruction* new_root_instruction,