Searched refs:branch_comp (Results 1 – 2 of 2) sorted by relevance
617 auto* branch_comp = conditional->branch_computation(branch); in Run() local618 if (!calling_conditionals.contains(branch_comp)) { in Run()619 calling_computationals_vector.push_back(branch_comp); in Run()621 calling_conditionals[branch_comp].insert(conditional); in Run()
880 auto branch_comp = instruction->branch_computation(operand_no - 1); in CopyOperandIfLayoutsDiffer() local881 auto param = branch_comp->parameter_instruction(0); in CopyOperandIfLayoutsDiffer()891 if (param == branch_comp->root_instruction()) { in CopyOperandIfLayoutsDiffer()892 branch_comp->set_root_instruction(param_copy, in CopyOperandIfLayoutsDiffer()895 *FindOrDie(computation_layouts_, branch_comp).mutable_parameter_layout(0) = in CopyOperandIfLayoutsDiffer()