Home
last modified time | relevance | path

Searched refs:FeatureID (Results 1 – 3 of 3) sorted by relevance

/llvm-project-15.0.7/llvm/lib/Analysis/
H A DTFUtils.cpp243 char *addNewTensor(size_t FeatureID) { in addNewTensor() argument
244 const auto &Spec = LoggedFeatureSpecs[FeatureID].Spec; in addNewTensor()
246 auto *RF = FeatureLists[FeatureID] in addNewTensor()
253 auto *RF = FeatureLists[FeatureID] in addNewTensor()
463 void Logger::logFloatValue(size_t FeatureID, const float *Value) {
464 assert(FeatureSpecs[FeatureID].Spec.isElementType<float>());
469 assert(FeatureSpecs[FeatureID].Spec.isElementType<int64_t>()); in logInt64Value()
474 assert(FeatureSpecs[FeatureID].Spec.isElementType<int32_t>()); in logInt32Value()
479 const auto &Spec = FeatureSpecs[FeatureID].Spec; in logSpecifiedTensorValue()
480 char *Buff = addEntryAndGetFloatOrInt64Buffer(FeatureID); in logSpecifiedTensorValue()
[all …]
/llvm-project-15.0.7/llvm/include/llvm/Analysis/Utils/
H A DTFUtils.h87 void logFloatValue(size_t FeatureID, const float *Value);
88 void logInt32Value(size_t FeatureID, const int32_t *Value);
89 void logInt64Value(size_t FeatureID, const int64_t *Value);
91 void logSpecifiedTensorValue(size_t FeatureID, const char *RawData);
98 char *addEntryAndGetFloatOrInt64Buffer(size_t FeatureID);
/llvm-project-15.0.7/llvm/include/llvm/Analysis/
H A DMLModelRunner.h35 template <typename T, typename I> T *getTensor(I FeatureID) { in getTensor() argument
37 getTensorUntyped(static_cast<size_t>(FeatureID))); in getTensor()
40 template <typename T, typename I> const T *getTensor(I FeatureID) const { in getTensor() argument
42 getTensorUntyped(static_cast<size_t>(FeatureID))); in getTensor()