Home
last modified time | relevance | path

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

/freebsd-14.2/contrib/llvm-project/llvm/include/llvm/Analysis/
H A DTensorSpec.h58 class TensorSpec final {
61 static TensorSpec createSpec(const std::string &Name,
64 return TensorSpec(Name, Port, getDataType<T>(), sizeof(T), Shape);
72 bool operator==(const TensorSpec &Other) const {
77 bool operator!=(const TensorSpec &Other) const { return !(*this == Other); }
90 TensorSpec(const std::string &NewName, const TensorSpec &Other) in TensorSpec() function
91 : TensorSpec(NewName, Other.Port, Other.Type, Other.ElementSize, in TensorSpec()
97 TensorSpec(const std::string &Name, int Port, TensorType Type,
111 std::string tensorValueToString(const char *Buffer, const TensorSpec &Spec);
120 std::optional<TensorSpec> getTensorSpecFromJSON(LLVMContext &Ctx,
[all …]
H A DModelUnderTrainingRunner.h38 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.h44 const std::vector<TensorSpec> &Inputs,
45 const TensorSpec &Advice, StringRef OutboundName,
63 const std::vector<TensorSpec> InputSpecs;
64 const TensorSpec OutputSpec;
H A DInlineModelFeatureMaps.h146 extern const std::vector<TensorSpec> FeatureMap;
149 extern const TensorSpec InlineDecisionSpec;
151 extern const TensorSpec DefaultDecisionSpec;
H A DNoInferenceModelRunner.h23 const std::vector<TensorSpec> &Inputs);
H A DMLModelRunner.h63 void setUpBufferForTensor(size_t Index, const TensorSpec &Spec, in setUpBufferForTensor()
/freebsd-14.2/contrib/llvm-project/llvm/lib/Analysis/
H A DModelUnderTrainingRunner.cpp25 TensorSpec Spec;
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()
95 const std::vector<TensorSpec> &InputSpecs, in ModelUnderTrainingRunner()
96 const std::vector<TensorSpec> &OutputSpecs, in ModelUnderTrainingRunner()
97 const std::vector<TensorSpec> &ExtraOutputsForLogging) in ModelUnderTrainingRunner()
125 const std::vector<TensorSpec> &InputSpecs, in createAndEnsureValid()
130 std::vector<TensorSpec> OutputSpecs; in createAndEnsureValid()
[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.cpp30 void Logger::writeHeader(std::optional<TensorSpec> AdviceSpec) { in writeHeader()
83 const std::vector<TensorSpec> &FeatureSpecs, in Logger()
84 const TensorSpec &RewardSpec, bool IncludeReward, in Logger()
85 std::optional<TensorSpec> AdviceSpec) in Logger()
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 DMLInlineAdvisor.cpp95 const std::vector<TensorSpec> llvm::FeatureMap{
96 #define POPULATE_NAMES(DTYPE, SHAPE, NAME, __) TensorSpec::createSpec<DTYPE>(#NAME, SHAPE),
107 const TensorSpec llvm::InlineDecisionSpec =
108 TensorSpec::createSpec<int64_t>(DecisionName, {1});
110 const TensorSpec llvm::DefaultDecisionSpec =
111 TensorSpec::createSpec<int64_t>(DefaultDecisionName, {1});
H A DNoInferenceModelRunner.cpp18 LLVMContext &Ctx, const std::vector<TensorSpec> &Inputs) in NoInferenceModelRunner()
H A DDevelopmentModeInlineAdvisor.cpp47 - "tensor_spec, followed by the TensorSpec description of the
/freebsd-14.2/contrib/llvm-project/llvm/include/llvm/Analysis/Utils/
H A DTrainingLogger.h93 const std::vector<TensorSpec> FeatureSpecs;
94 const TensorSpec RewardSpec;
99 void writeHeader(std::optional<TensorSpec> AdviceSpec);
100 void writeTensor(const TensorSpec &Spec, const char *RawData) { in writeTensor()
113 const std::vector<TensorSpec> &FeatureSpecs,
114 const TensorSpec &RewardSpec, bool IncludeReward,
115 std::optional<TensorSpec> AdviceSpec = std::nullopt);
H A DTFUtils.h77 const std::vector<TensorSpec> &InputSpecs,
78 const std::vector<TensorSpec> &OutputSpecs,
/freebsd-14.2/contrib/llvm-project/llvm/lib/CodeGen/
H A DMLRegAllocPriorityAdvisor.cpp79 static const TensorSpec DecisionSpec =
80 TensorSpec::createSpec<float>(DecisionName, {1});
113 TensorSpec::createSpec<type>(#name, shape),
115 static const std::vector<TensorSpec> InputFeatures{
164 static const TensorSpec Reward = TensorSpec::createSpec<float>("reward", {1});
167 TensorSpec::createSpec<type>(std::string("action_") + #name, shape),
169 static const std::vector<TensorSpec> TrainingInputFeatures{
171 TensorSpec::createSpec<float>("action_discount", {1}),
172 TensorSpec::createSpec<int32_t>("action_step_type", {1}),
173 TensorSpec::createSpec<float>("action_reward", {1})}};
[all …]
H A DMLRegAllocEvictAdvisor.cpp226 static const TensorSpec DecisionSpec =
227 TensorSpec::createSpec<int64_t>(DecisionName, {1});
363 TensorSpec::createSpec<type>(#name, shape),
387 std::vector<TensorSpec> InputFeatures;
420 static const TensorSpec Reward = TensorSpec::createSpec<float>("reward", {1});
460 TensorSpec::createSpec<int32_t>("action_step_type", {1}), in DevelopmentModeEvictionAdvisorAnalysis()
461 TensorSpec::createSpec<float>("action_reward", {1})}; in DevelopmentModeEvictionAdvisorAnalysis()
468 TensorSpec::createSpec<float>("action_reward", {1})}; in DevelopmentModeEvictionAdvisorAnalysis()
493 std::vector<TensorSpec> InputFeatures;
494 std::vector<TensorSpec> TrainingInputFeatures;
[all …]
/freebsd-14.2/lib/clang/libllvm/
H A DMakefile156 SRCS_MIN+= Analysis/TensorSpec.cpp