Searched refs:induction_vars (Results 1 – 4 of 4) sorted by relevance
790 absl::flat_hash_map<int64, const HloConstantInstruction*> induction_vars; in TryMergeInductionVariables() local806 induction_vars.emplace(i, Cast<HloConstantInstruction>(constant)); in TryMergeInductionVariables()817 if (induction_vars.size() + (trip_counter.has_value() ? 1 : 0) < 2) { in TryMergeInductionVariables()871 if (!induction_vars.count(i)) { in TryMergeInductionVariables()879 add_new_instr(induction_vars.at(i)->Clone())))); in TryMergeInductionVariables()897 add_gte((induction_vars.count(i) ? loop_body_param : old_root), i)); in TryMergeInductionVariables()
35 LoopVariableOptimizer* induction_vars);
59 explicit Visitor(Typer* typer, LoopVariableOptimizer* induction_vars) in Visitor() argument61 induction_vars_(induction_vars), in Visitor()349 LoopVariableOptimizer* induction_vars) { in Run() argument350 if (induction_vars != nullptr) { in Run()351 induction_vars->ChangeToInductionVariablePhis(); in Run()353 Visitor visitor(this, induction_vars); in Run()359 if (induction_vars != nullptr) { in Run()360 induction_vars->ChangeToPhisAndInsertGuards(); in Run()
1245 LoopVariableOptimizer induction_vars(data->jsgraph()->graph(), in Run() local1247 if (FLAG_turbo_loop_variable) induction_vars.Run(); in Run()1248 typer->Run(roots, &induction_vars); in Run()