Searched refs:branch_i (Results 1 – 1 of 1) sorted by relevance
1306 HloComputation* branch_i = instr->branch_computation(i); in Run() local1307 if (ContainsKey(conditional_computations, branch_i)) { in Run()1308 conditional_computations[branch_i]++; in Run()1310 conditional_computations[branch_i] = 0; in Run()1338 HloComputation* branch_i = conditional->branch_computation(i); in Run() local1339 if (conditional_computations[branch_i] > 0) { in Run()1409 HloComputation* branch_i = conditional->branch_computation(i); in Run() local1410 if (conditional_computations[branch_i] > 0) { in Run()1422 branch_i->Clone("clone", &clone_context)); in Run()1424 conditional_computations[branch_i]--; in Run()