Home
last modified time | relevance | path

Searched refs:TensorSpec (Results 1 – 23 of 23) sorted by relevance

/freebsd/contrib/llvm-project/llvm/include/llvm/Analysis/
H A DTensorSpec.h63 class TensorSpec final {
66 static TensorSpec createSpec(const std::string &Name,
69 return TensorSpec(Name, Port, getDataType<T>(), sizeof(T), Shape);
77 bool operator==(const TensorSpec &Other) const {
82 bool operator!=(const TensorSpec &Other) const { return !(*this == Other); }
95 TensorSpec(const std::string &NewName, const TensorSpec &Other) in TensorSpec() function
96 : TensorSpec(NewName, Other.Port, Other.Type, Other.ElementSize, in TensorSpec()
102 LLVM_ABI TensorSpec(const std::string &Name, int Port, TensorType Type,
117 const TensorSpec &Spec);
126 LLVM_ABI std::optional<TensorSpec>
[all …]
H A DModelUnderTrainingRunner.h15 #include "llvm/Analysis/TensorSpec.h"
38 const std::vector<TensorSpec> &extraOutputsForLoggingSpecs() const { in extraOutputsForLoggingSpecs()
57 const std::vector<TensorSpec> &InputSpecs,
62 const std::vector<TensorSpec> &InputSpecs,
63 const std::vector<TensorSpec> &OutputSpecs,
64 const std::vector<TensorSpec> &ExtraOutputsForLogging = {});
70 const std::vector<TensorSpec> OutputSpecs;
71 const std::vector<TensorSpec> ExtraOutputsForLogging;
H A DInteractiveModelRunner.h42 const std::vector<TensorSpec> &Inputs,
43 const TensorSpec &Advice, StringRef OutboundName,
61 const std::vector<TensorSpec> InputSpecs;
62 const TensorSpec OutputSpec;
H A DNoInferenceModelRunner.h16 class TensorSpec; variable
24 const std::vector<TensorSpec> &Inputs);
H A DInlineModelFeatureMaps.h163 LLVM_ABI extern std::vector<TensorSpec> FeatureMap;
166 LLVM_ABI extern const TensorSpec InlineDecisionSpec;
168 LLVM_ABI extern const TensorSpec DefaultDecisionSpec;
H A DReleaseModeModelRunner.h74 TensorSpec::createSpec<uint64_t>("model_selector", {2}),
119 void populateTensor(size_t Pos, const TensorSpec &Spec, StringRef Prefix, in populateTensor()
H A DMLModelRunner.h63 void setUpBufferForTensor(size_t Index, const TensorSpec &Spec, in setUpBufferForTensor()
/freebsd/contrib/llvm-project/llvm/lib/Analysis/
H A DModelUnderTrainingRunner.cpp25 TensorSpec Spec;
60 if (auto TensorSpec = getTensorSpecFromJSON(Ctx, *SpecPart)) in loadOutputSpecs() local
62 if (!TensorSpec->isElementType<int64_t>() && in loadOutputSpecs()
63 !TensorSpec->isElementType<int32_t>() && in loadOutputSpecs()
64 !TensorSpec->isElementType<float>()) { in loadOutputSpecs()
68 TensorSpec->name()); in loadOutputSpecs()
71 Ret.push_back({*TensorSpec, LoggingName->str()}); in loadOutputSpecs()
95 const std::vector<TensorSpec> &InputSpecs, in ModelUnderTrainingRunner()
96 const std::vector<TensorSpec> &OutputSpecs, in ModelUnderTrainingRunner()
97 const std::vector<TensorSpec> &ExtraOutputsForLogging) in ModelUnderTrainingRunner()
[all …]
H A DTensorSpec.cpp33 template <> TensorType TensorSpec::getDataType<T>() { return TensorType::E; }
50 void TensorSpec::toJSON(json::OStream &OS) const { in toJSON()
62 TensorSpec::TensorSpec(const std::string &Name, int Port, TensorType Type, in TensorSpec() function in llvm::TensorSpec
69 std::optional<TensorSpec> getTensorSpecFromJSON(LLVMContext &Ctx, in getTensorSpecFromJSON()
72 [&](const llvm::Twine &Message) -> std::optional<TensorSpec> { in getTensorSpecFromJSON()
101 return TensorSpec::createSpec<T>(TensorName, TensorShape, TensorPort); in getTensorSpecFromJSON()
107 std::string tensorValueToString(const char *Buffer, const TensorSpec &Spec) { in tensorValueToString()
H A DTFLiteUtils.cpp56 const std::vector<TensorSpec> &InputSpecs,
57 const std::vector<TensorSpec> &OutputSpecs,
95 const TensorSpec &Spec);
101 StringRef SavedModelPath, const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluatorImpl()
102 const std::vector<TensorSpec> &OutputSpecs, const char *Tags = "serve") in TFModelEvaluatorImpl()
185 const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluator()
186 const std::vector<TensorSpec> &OutputSpecs, in TFModelEvaluator()
197 const TensorSpec &Spec) { in checkReportAndInvalidate()
H A DTrainingLogger.cpp13 #include "llvm/Analysis/TensorSpec.h"
30 void Logger::writeHeader(std::optional<TensorSpec> AdviceSpec) {
83 const std::vector<TensorSpec> &FeatureSpecs, in logRewardImpl()
84 const TensorSpec &RewardSpec, bool IncludeReward, in logRewardImpl()
85 std::optional<TensorSpec> AdviceSpec) in logRewardImpl()
H A DMLInlineAdvisor.cpp110 std::vector<TensorSpec> llvm::FeatureMap{
111 #define POPULATE_NAMES(DTYPE, SHAPE, NAME, __) TensorSpec::createSpec<DTYPE>(#NAME, SHAPE),
122 const TensorSpec llvm::InlineDecisionSpec =
123 TensorSpec::createSpec<int64_t>(DecisionName, {1});
125 const TensorSpec llvm::DefaultDecisionSpec =
126 TensorSpec::createSpec<int64_t>(DefaultDecisionName, {1});
199 TensorSpec::createSpec<float>("callee_embedding", {IR2VecDim})); in MLInlineAdvisor()
201 TensorSpec::createSpec<float>("caller_embedding", {IR2VecDim})); in MLInlineAdvisor()
H A DInlineSizeEstimatorAnalysis.cpp222 std::vector<TensorSpec> InputSpecs{TensorSpec::createSpec<int32_t>( in InlineSizeEstimatorAnalysis()
226 std::vector<TensorSpec> OutputSpecs{ in InlineSizeEstimatorAnalysis()
227 TensorSpec::createSpec<float>("StatefulPartitionedCall", {1})}; in InlineSizeEstimatorAnalysis()
H A DInteractiveModelRunner.cpp27 LLVMContext &Ctx, const std::vector<TensorSpec> &Inputs, in InteractiveModelRunner()
28 const TensorSpec &Advice, StringRef OutboundName, StringRef InboundName) in InteractiveModelRunner()
H A DNoInferenceModelRunner.cpp18 LLVMContext &Ctx, const std::vector<TensorSpec> &Inputs) in NoInferenceModelRunner()
H A DDevelopmentModeInlineAdvisor.cpp48 - "tensor_spec, followed by the TensorSpec description of the
/freebsd/contrib/llvm-project/llvm/include/llvm/Analysis/Utils/
H A DTrainingLogger.h94 const std::vector<TensorSpec> FeatureSpecs;
95 const TensorSpec RewardSpec;
100 void writeHeader(std::optional<TensorSpec> AdviceSpec);
101 void writeTensor(const TensorSpec &Spec, const char *RawData) { in writeTensor()
114 const std::vector<TensorSpec> &FeatureSpecs,
115 const TensorSpec &RewardSpec, bool IncludeReward,
116 std::optional<TensorSpec> AdviceSpec = std::nullopt);
H A DTFUtils.h16 #include "llvm/Analysis/TensorSpec.h"
77 const std::vector<TensorSpec> &InputSpecs,
78 const std::vector<TensorSpec> &OutputSpecs,
/freebsd/contrib/llvm-project/llvm/lib/CodeGen/
H A DMLRegallocPriorityAdvisor.cpp
H A DMLRegAllocPriorityAdvisor.cpp80 static const TensorSpec DecisionSpec =
81 TensorSpec::createSpec<float>(DecisionName, {1});
114 TensorSpec::createSpec<type>(#name, shape),
116 static const std::vector<TensorSpec> InputFeatures{
178 static const TensorSpec Reward = TensorSpec::createSpec<float>("reward", {1});
181 TensorSpec::createSpec<type>(std::string("action_") + #name, shape),
183 static const std::vector<TensorSpec> TrainingInputFeatures{
185 TensorSpec::createSpec<float>("action_discount", {1}),
186 TensorSpec::createSpec<int32_t>("action_step_type", {1}),
187 TensorSpec::createSpec<float>("action_reward", {1})}};
[all …]
H A DMLRegallocEvictAdvisor.cpp
H A DMLRegAllocEvictAdvisor.cpp234 static const TensorSpec DecisionSpec =
235 TensorSpec::createSpec<int64_t>(DecisionName, {1});
387 TensorSpec::createSpec<type>(#name, shape),
431 std::vector<TensorSpec> InputFeatures;
469 static const TensorSpec Reward = TensorSpec::createSpec<float>("reward", {1});
476 TensorSpec::createSpec<type>(std::string("action_") + #name, shape),
508 TensorSpec::createSpec<float>("action_discount", {1}), in DevelopmentModeEvictionAdvisorProvider()
509 TensorSpec::createSpec<int32_t>("action_step_type", {1}), in DevelopmentModeEvictionAdvisorProvider()
510 TensorSpec::createSpec<float>("action_reward", {1})}; in DevelopmentModeEvictionAdvisorProvider()
515 TensorSpec::createSpec<float>("action_discount", {1}), in DevelopmentModeEvictionAdvisorProvider()
[all …]
/freebsd/lib/clang/libllvm/
H A DMakefile184 SRCS_MIN+= Analysis/TensorSpec.cpp