Searched refs:conditional_computations (Results 1 – 1 of 1) sorted by relevance
1417 absl::flat_hash_map<HloComputation*, int> conditional_computations; in Run() local1424 if (ContainsKey(conditional_computations, branch_i)) { in Run()1425 conditional_computations[branch_i]++; in Run()1427 conditional_computations[branch_i] = 0; in Run()1470 if (conditional_computations[branch_i] > 0) { in Run()1541 if (conditional_computations[branch_i] > 0) { in Run()1555 conditional_computations[branch_i]--; in Run()