Searched refs:model_parallelism (Results 1 – 2 of 2) sorted by relevance
173 // CHECK-LABEL: func @model_parallelism174 func.func @model_parallelism() -> () {
2261 int64_t model_parallelism = 0; in ShouldStop() local2263 model_parallelism += std::round(pair.second->value); in ShouldStop()2265 *cpu_budget_reached = (model_parallelism > cpu_budget); in ShouldStop()