Home
last modified time | relevance | path

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

/freebsd/contrib/llvm-project/llvm/lib/Analysis/
H A DMLInlineAdvisor.cpp424 *ModelRunner->getTensor<int64_t>(FeatureIndex::callee_basic_block_count) = in getAdviceImpl()
426 *ModelRunner->getTensor<int64_t>(FeatureIndex::callsite_height) = in getAdviceImpl()
428 *ModelRunner->getTensor<int64_t>(FeatureIndex::node_count) = NodeCount; in getAdviceImpl()
429 *ModelRunner->getTensor<int64_t>(FeatureIndex::nr_ctant_params) = in getAdviceImpl()
431 *ModelRunner->getTensor<int64_t>(FeatureIndex::edge_count) = EdgeCount; in getAdviceImpl()
432 *ModelRunner->getTensor<int64_t>(FeatureIndex::caller_users) = in getAdviceImpl()
434 *ModelRunner->getTensor<int64_t>( in getAdviceImpl()
437 *ModelRunner->getTensor<int64_t>(FeatureIndex::caller_basic_block_count) = in getAdviceImpl()
439 *ModelRunner->getTensor<int64_t>( in getAdviceImpl()
442 *ModelRunner->getTensor<int64_t>(FeatureIndex::callee_users) = in getAdviceImpl()
[all …]
/freebsd/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;
/freebsd/contrib/llvm-project/llvm/lib/CodeGen/
H A DMLRegallocPriorityAdvisor.cpp
H A DMLRegAllocPriorityAdvisor.cpp332 *Runner->getTensor<int64_t>(0) = static_cast<int64_t>(Size); in getPriorityImpl()
333 *Runner->getTensor<int64_t>(1) = static_cast<int64_t>(Stage); in getPriorityImpl()
334 *Runner->getTensor<float>(2) = static_cast<float>(LI.weight()); in getPriorityImpl()
H A DMLRegallocEvictAdvisor.cpp
H A DMLRegAllocEvictAdvisor.cpp850 Runner->getTensor<float>(FeatureIndex)[Pos] /= Largest[FeatureIndex]; in tryFindEvictionCandidate()
853 *Runner->getTensor<float>(FeatureIDs::progress) = in tryFindEvictionCandidate()
1008 Runner->getTensor<TYPE>(FeatureIDs::ID)[Pos] = static_cast<TYPE>(VAL); \ in extractFeatures()
1101 RegallocRunner->getTensor<int64_t>(InstructionsIndex)[InstructionIndex] = in extractInstructionFeatures()
1105 RegallocRunner->getTensor<int64_t>( in extractInstructionFeatures()
1123 RegallocRunner->getTensor<int64_t>( in extractInstructionFeatures()
1162 RegallocRunner->getTensor<float>(MBBFreqIndex)[CurrentMBBIndex] = in extractMBBFrequency()
1164 RegallocRunner->getTensor<int64_t>( in extractMBBFrequency()