Searched refs:conditional_computations (Results 1 – 1 of 1) sorted by relevance
1319 absl::flat_hash_map<HloComputation*, int> conditional_computations; in Run() local1326 if (ContainsKey(conditional_computations, branch_i)) { in Run()1327 conditional_computations[branch_i]++; in Run()1329 conditional_computations[branch_i] = 0; in Run()1372 if (conditional_computations[branch_i] > 0) { in Run()1443 if (conditional_computations[branch_i] > 0) { in Run()1457 conditional_computations[branch_i]--; in Run()