Searched refs:new_root_instruction (Results 1 – 2 of 2) sorted by relevance
330 void HloComputation::set_root_instruction(HloInstruction* new_root_instruction, in set_root_instruction() argument335 CHECK(ShapeUtil::Compatible(new_root_instruction->shape(), in set_root_instruction()337 << new_root_instruction->shape() << " is incompatible with " in set_root_instruction()342 if (new_root_instruction == instruction.get()) { in set_root_instruction()351 if (!Shape::Equal().IgnoreLayout()(new_root_instruction->shape(), in set_root_instruction()355 HloInputOutputAliasConfig(new_root_instruction->shape()); in set_root_instruction()359 root_instruction_ = new_root_instruction; in set_root_instruction()
185 void set_root_instruction(HloInstruction* new_root_instruction,