Searched refs:conditional_computations (Results 1 – 1 of 1) sorted by relevance
1300 absl::flat_hash_map<HloComputation*, int> conditional_computations; 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()1339 if (conditional_computations[branch_i] > 0) { in Run()1410 if (conditional_computations[branch_i] > 0) { in Run()1424 conditional_computations[branch_i]--; in Run()