Searched refs:cpu_budget_reached (Results 1 – 2 of 2) sorted by relevance
1700 std::shared_ptr<Node> snapshot, bool* cpu_budget_reached) { in ShouldStop() argument1701 if (!(*cpu_budget_reached)) { in ShouldStop()1709 *cpu_budget_reached = (model_parallelism > cpu_budget); in ShouldStop()1714 (*cpu_budget_reached ? buffer_size_parameters : parameters)) { in ShouldStop()1810 bool cpu_budget_reached = false; in OptimizeGradientDescent() local1816 snapshot, &cpu_budget_reached); in OptimizeGradientDescent()1828 gradients, &(cpu_budget_reached ? buffer_size_parameters : parameters)); in OptimizeGradientDescent()
757 std::shared_ptr<Node> snapshot, bool* cpu_budget_reached);