Searched refs:optimization_params (Results 1 – 4 of 4) sorted by relevance
1646 OptimizationParams optimization_params; in Optimize() local1647 optimization_params.set_algorithm(algorithm); in Optimize()1648 optimization_params.set_cpu_budget(cpu_budget); in Optimize()1649 optimization_params.set_ram_budget(ram_budget); in Optimize()1650 optimization_params.set_model_input_time(model_input_time); in Optimize()1653 OptimizeHillClimb(snapshot, optimization_params); in Optimize()1656 OptimizeGradientDescent(snapshot, optimization_params); in Optimize()1667 save_buffer_.push_back(std::make_pair(snapshot, optimization_params)); in Optimize()1777 const OptimizationParams& optimization_params) { in OptimizeGradientDescent() argument1813 !ShouldStop(optimization_params.cpu_budget(), in OptimizeGradientDescent()[all …]
945 ModelProto::OptimizationParams optimization_params; in TEST() local946 optimization_params.set_algorithm(AutotuneAlgorithm::GRADIENT_DESCENT); in TEST()947 optimization_params.set_cpu_budget(64); in TEST()948 optimization_params.set_ram_budget(1024); in TEST()949 optimization_params.set_model_input_time(43653.34534); in TEST()951 model.output()->Snapshot(), optimization_params)); in TEST()959 EXPECT_EQ(optimization_params.algorithm(), in TEST()961 EXPECT_EQ(optimization_params.cpu_budget(), in TEST()963 EXPECT_EQ(optimization_params.ram_budget(), in TEST()965 EXPECT_EQ(optimization_params.model_input_time(), in TEST()
699 const OptimizationParams& optimization_params);704 OptimizationParams* optimization_params);729 const OptimizationParams& optimization_params);739 const OptimizationParams& optimization_params);
129 OptimizationParams optimization_params = 4; field