Searched refs:select_root (Results 1 – 1 of 1) sorted by relevance
3168 auto select_root = select->root_instruction(); in HandleSelectAndScatter() local3169 if (select_root->opcode() != HloOpcode::kCompare || in HandleSelectAndScatter()3170 select_root->operand(0)->opcode() != HloOpcode::kParameter || in HandleSelectAndScatter()3171 select_root->operand(1)->opcode() != HloOpcode::kParameter || in HandleSelectAndScatter()3172 select_root->operand(0)->parameter_number() + in HandleSelectAndScatter()3173 select_root->operand(1)->parameter_number() != in HandleSelectAndScatter()3179 if (select_root->comparison_direction() == ComparisonDirection::kGe || in HandleSelectAndScatter()3180 select_root->comparison_direction() == ComparisonDirection::kGt) { in HandleSelectAndScatter()3181 if (select_root->operand(0)->parameter_number() == 0) { in HandleSelectAndScatter()3186 } else if (select_root->comparison_direction() == ComparisonDirection::kLe || in HandleSelectAndScatter()[all …]