Home
last modified time | relevance | path

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

/llvm-project-15.0.7/llvm/unittests/Analysis/
H A DMLModelRunnerTest.cpp55 const std::vector<TensorSpec> Inputs{ in TEST()
56 TensorSpec::createSpec<int64_t>("F1", {1}), in TEST()
57 TensorSpec::createSpec<int64_t>("F2", {10}), in TEST()
58 TensorSpec::createSpec<float>("F2", {5}), in TEST()
76 std::vector<TensorSpec> Inputs{TensorSpec::createSpec<int64_t>("a", {1}), in TEST()
89 std::vector<TensorSpec> Inputs{TensorSpec::createSpec<int64_t>("a", {1}), in TEST()
90 TensorSpec::createSpec<int64_t>("b", {1}), in TEST()
105 std::vector<TensorSpec> Inputs{ in TEST()
106 TensorSpec::createSpec<int64_t>("a", {1}), in TEST()
107 TensorSpec::createSpec<int64_t>("c", {1}), in TEST()
[all …]
H A DTFUtilsTest.cpp48 std::vector<TensorSpec> InputSpecs{TensorSpec::createSpec<int32_t>( in TEST()
50 std::vector<TensorSpec> OutputSpecs{ in TEST()
89 std::vector<TensorSpec> InputSpecs{TensorSpec::createSpec<int32_t>( in TEST()
91 std::vector<TensorSpec> OutputSpecs{ in TEST()
109 std::vector<TensorSpec> InputSpecs{ in TEST()
110 TensorSpec::createSpec<int32_t>("serving_default_input_1", in TEST()
156 auto Rewards = TensorSpec::createSpec<float>("reward", {1}); in TEST()
192 auto Rewards = TensorSpec::createSpec<int32_t>("reward", {1}); in TEST()
228 auto Rewards = TensorSpec::createSpec<float>("reward", {1}); in TEST()
256 auto Rewards = TensorSpec::createSpec<float>("reward", {1}); in TEST()
[all …]
H A DTensorSpecTest.cpp28 Optional<TensorSpec> Spec = getTensorSpecFromJSON(Ctx, *Value); in TEST()
30 EXPECT_EQ(*Spec, TensorSpec::createSpec<int32_t>("tensor_name", {1, 4}, 2)); in TEST()
49 auto Spec1D = TensorSpec::createSpec<int16_t>("Hi1", {1}); in TEST()
50 auto Spec2D = TensorSpec::createSpec<int16_t>("Hi2", {1, 1}); in TEST()
51 auto Spec1DLarge = TensorSpec::createSpec<float>("Hi3", {10}); in TEST()
52 auto Spec3DLarge = TensorSpec::createSpec<float>("Hi3", {2, 4, 10}); in TEST()
/llvm-project-15.0.7/llvm/lib/Analysis/
H A DTensorSpec.cpp32 template <> TensorType TensorSpec::getDataType<T>() { return TensorType::E; }
38 TensorSpec::TensorSpec(const std::string &Name, int Port, TensorType Type, in SUPPORTED_TENSOR_TYPES()
45 Optional<TensorSpec> getTensorSpecFromJSON(LLVMContext &Ctx, in getTensorSpecFromJSON()
47 auto EmitError = [&](const llvm::Twine &Message) -> Optional<TensorSpec> { in getTensorSpecFromJSON()
76 return TensorSpec::createSpec<T>(TensorName, TensorShape, TensorPort); in getTensorSpecFromJSON()
113 if (auto TensorSpec = getTensorSpecFromJSON(Ctx, *SpecPart)) in loadOutputSpecs() local
115 if (!TensorSpec->isElementType<int64_t>() && in loadOutputSpecs()
116 !TensorSpec->isElementType<int32_t>() && in loadOutputSpecs()
117 !TensorSpec->isElementType<float>()) { in loadOutputSpecs()
121 TensorSpec->name()); in loadOutputSpecs()
[all …]
H A DTFUtils.cpp135 const std::vector<TensorSpec> &InputSpecs,
136 function_ref<TensorSpec(size_t)> GetOutputSpecs,
182 const TensorSpec &OutputSpec);
187 const TensorSpec RewardSpec;
229 const TensorSpec &RewardSpec, bool IncludeReward) in LoggerDataImpl()
281 StringRef SavedModelPath, const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluatorImpl()
282 function_ref<TensorSpec(size_t)> GetOutputSpecs, size_t OutputSpecsSize, in TFModelEvaluatorImpl()
334 StringRef SavedModelPath, const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluator()
335 function_ref<TensorSpec(size_t)> GetOutputSpecs, size_t OutputSpecsSize, in TFModelEvaluator()
365 const TF_Output &Output, const TensorSpec &OutputSpec) { in checkReportAndInvalidate()
[all …]
H A DModelUnderTrainingRunner.cpp23 const std::vector<TensorSpec> &InputSpecs, in ModelUnderTrainingRunner()
53 const std::vector<TensorSpec> &InputSpecs, in createAndEnsureValid()
66 const std::vector<TensorSpec> &InputSpecs, in createAndEnsureValid()
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 DNoInferenceModelRunner.cpp18 LLVMContext &Ctx, const std::vector<TensorSpec> &Inputs) in NoInferenceModelRunner()
H A DDevelopmentModeInlineAdvisor.cpp44 - "tensor_spec, followed by the TensorSpec description of the
H A DMLInlineAdvisor.cpp62 const std::array<TensorSpec, NumberOfFeatures> llvm::FeatureMap{
63 #define POPULATE_NAMES(_, NAME) TensorSpec::createSpec<int64_t>(NAME, {1} ),
69 #define POPULATE_NAMES(_, NAME, __) TensorSpec::createSpec<int64_t>(NAME, {1} ),
H A DCMakeLists.txt132 TensorSpec.cpp
/llvm-project-15.0.7/llvm/include/llvm/Analysis/
H A DTensorSpec.h52 class TensorSpec final {
55 static TensorSpec createSpec(const std::string &Name,
58 return TensorSpec(Name, Port, getDataType<T>(), sizeof(T), Shape);
66 bool operator==(const TensorSpec &Other) const {
71 bool operator!=(const TensorSpec &Other) const { return !(*this == Other); }
85 TensorSpec(const std::string &Name, int Port, TensorType Type,
105 Optional<TensorSpec> getTensorSpecFromJSON(LLVMContext &Ctx,
109 TensorSpec Spec;
126 template <> TensorType TensorSpec::getDataType<T>();
H A DModelUnderTrainingRunner.h50 const std::vector<TensorSpec> &InputSpecs,
55 const std::vector<TensorSpec> &InputSpecs,
60 const std::vector<TensorSpec> &InputSpecs,
H A DNoInferenceModelRunner.h23 const std::vector<TensorSpec> &Inputs);
H A DMLModelRunner.h60 void setUpBufferForTensor(size_t Index, const TensorSpec &Spec, in setUpBufferForTensor()
H A DInlineModelFeatureMaps.h132 extern const std::array<TensorSpec, NumberOfFeatures> FeatureMap;
/llvm-project-15.0.7/llvm/include/llvm/Analysis/Utils/
H A DTFUtils.h75 const TensorSpec &RewardSpec, bool IncludeReward);
112 TensorSpec RewardSpec;
152 const std::vector<TensorSpec> &InputSpecs,
153 const std::vector<TensorSpec> &OutputSpecs,
156 const std::vector<TensorSpec> &InputSpecs,
157 function_ref<TensorSpec(size_t)> GetOutputSpecs,
/llvm-project-15.0.7/llvm/lib/Analysis/models/
H A Dgen-inline-oz-test-model.py37 tf.TensorSpec(dtype=tf.int64, shape=(), name=key) for key in [
79 tf.TensorSpec(dtype=tf.float32, shape=(), name=key)
85 tf.TensorSpec(dtype=tf.int32, shape=(), name=key)
H A Dgen-regalloc-eviction-test-model.py32 (key, tf.TensorSpec(dtype=tf.int64, shape=(NUM_REGISTERS), name=key))
/llvm-project-15.0.7/llvm/lib/CodeGen/
H A DMLRegallocEvictAdvisor.cpp323 TensorSpec::createSpec<type>(#name, shape),
325 static const std::vector<TensorSpec> InputFeatures{
367 static const TensorSpec Output =
368 TensorSpec::createSpec<int64_t>(DecisionName, {1});
369 static const TensorSpec Reward = TensorSpec::createSpec<float>("reward", {1});
376 TensorSpec::createSpec<type>(std::string("action_") + #name, shape),
378 static const std::vector<TensorSpec> TrainingInputFeatures{
380 TensorSpec::createSpec<float>("action_discount", {1}),
381 TensorSpec::createSpec<int32_t>("action_step_type", {1}),
382 TensorSpec::createSpec<float>("action_reward", {1})}};
/llvm-project-15.0.7/llvm/utils/gn/secondary/llvm/lib/Analysis/
H A DBUILD.gn124 "TensorSpec.cpp",