Home
last modified time | relevance | path

Searched refs:ModelRunner (Results 1 – 2 of 2) sorted by relevance

/src/contrib/llvm-project/llvm/lib/Analysis/
H A DMLInlineAdvisor.cpp145 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()
[all …]
/src/contrib/llvm-project/llvm/include/llvm/Analysis/
H A DMLInlineAdvisor.h32 std::unique_ptr<MLModelRunner> ModelRunner,
48 const MLModelRunner &getModelRunner() const { return *ModelRunner; } in getModelRunner()
67 std::unique_ptr<MLModelRunner> ModelRunner; variable