Home
last modified time | relevance | path

Searched refs:getTensor (Results 1 – 5 of 5) sorted by relevance

/src/contrib/llvm-project/llvm/lib/Analysis/
H A DMLInlineAdvisor.cpp412 *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()
[all …]
/src/contrib/llvm-project/llvm/include/llvm/Analysis/
H A DMLModelRunner.h37 template <typename T, typename I> T *getTensor(I FeatureID) { in getTensor() function
42 template <typename T, typename I> const T *getTensor(I FeatureID) const { in getTensor() function
H A DReleaseModeModelRunner.h94 getTensor<uint64_t>(InputSpec.size())[0] = High;
95 getTensor<uint64_t>(InputSpec.size())[1] = Low;
/src/contrib/llvm-project/llvm/lib/CodeGen/
H A DMLRegAllocPriorityAdvisor.cpp301 *Runner->getTensor<int64_t>(0) = static_cast<int64_t>(Size); in getPriorityImpl()
302 *Runner->getTensor<int64_t>(1) = static_cast<int64_t>(Stage); in getPriorityImpl()
303 *Runner->getTensor<float>(2) = static_cast<float>(LI.weight()); in getPriorityImpl()
H A DMLRegAllocEvictAdvisor.cpp773 Runner->getTensor<float>(FeatureIndex)[Pos] /= Largest[FeatureIndex]; in tryFindEvictionCandidate()
776 *Runner->getTensor<float>(FeatureIDs::progress) = in tryFindEvictionCandidate()
918 Runner->getTensor<TYPE>(FeatureIDs::ID)[Pos] = static_cast<TYPE>(VAL); \ in extractFeatures()
1011 RegallocRunner->getTensor<int64_t>(InstructionsIndex)[InstructionIndex] = in extractInstructionFeatures()
1015 RegallocRunner->getTensor<int64_t>( in extractInstructionFeatures()
1033 RegallocRunner->getTensor<int64_t>( in extractInstructionFeatures()
1073 RegallocRunner->getTensor<float>(MBBFreqIndex)[CurrentMBBIndex] = in extractMBBFrequency()
1075 RegallocRunner->getTensor<int64_t>( in extractMBBFrequency()