Lines Matching refs:ModelRunner
145 ModelRunner(std::move(Runner)), GetDefaultAdvice(GetDefaultAdvice), in MLInlineAdvisor()
149 assert(ModelRunner); in MLInlineAdvisor()
150 ModelRunner->switchContext(""); in MLInlineAdvisor()
412 *ModelRunner->getTensor<int64_t>(FeatureIndex::callee_basic_block_count) = in getAdviceImpl()
414 *ModelRunner->getTensor<int64_t>(FeatureIndex::callsite_height) = in getAdviceImpl()
416 *ModelRunner->getTensor<int64_t>(FeatureIndex::node_count) = NodeCount; in getAdviceImpl()
417 *ModelRunner->getTensor<int64_t>(FeatureIndex::nr_ctant_params) = in getAdviceImpl()
419 *ModelRunner->getTensor<int64_t>(FeatureIndex::edge_count) = EdgeCount; in getAdviceImpl()
420 *ModelRunner->getTensor<int64_t>(FeatureIndex::caller_users) = in getAdviceImpl()
422 *ModelRunner->getTensor<int64_t>( in getAdviceImpl()
425 *ModelRunner->getTensor<int64_t>(FeatureIndex::caller_basic_block_count) = in getAdviceImpl()
427 *ModelRunner->getTensor<int64_t>( in getAdviceImpl()
430 *ModelRunner->getTensor<int64_t>(FeatureIndex::callee_users) = in getAdviceImpl()
432 *ModelRunner->getTensor<int64_t>(FeatureIndex::cost_estimate) = CostEstimate; in getAdviceImpl()
433 *ModelRunner->getTensor<int64_t>(FeatureIndex::is_callee_avail_external) = in getAdviceImpl()
435 *ModelRunner->getTensor<int64_t>(FeatureIndex::is_caller_avail_external) = in getAdviceImpl()
441 *ModelRunner->getTensor<int64_t>(inlineCostFeatureToMlFeature( in getAdviceImpl()
446 *ModelRunner->getTensor<int64_t>(InlineCostFeatureIndex::NumberOfFeatures) = in getAdviceImpl()
455 this, CB, ORE, static_cast<bool>(ModelRunner->evaluate<int64_t>())); in getAdviceFromModel()