Home
last modified time | relevance | path

Searched refs:InputSpecs (Results 1 – 6 of 6) sorted by relevance

/llvm-project-15.0.7/llvm/lib/Analysis/
H A DModelUnderTrainingRunner.cpp23 const std::vector<TensorSpec> &InputSpecs, in ModelUnderTrainingRunner() argument
25 : MLModelRunner(Ctx, MLModelRunner::Kind::Development, InputSpecs.size()), in ModelUnderTrainingRunner()
28 ModelPath, InputSpecs, [&](size_t I) { return OutputSpecs[I].Spec; }, in ModelUnderTrainingRunner()
36 for (size_t I = 0, E = InputSpecs.size(); I < E; ++I) { in ModelUnderTrainingRunner()
37 setUpBufferForTensor(I, InputSpecs[I], Evaluator->getUntypedInput(I)); in ModelUnderTrainingRunner()
53 const std::vector<TensorSpec> &InputSpecs, in createAndEnsureValid() argument
57 return createAndEnsureValid(Ctx, ModelPath, DecisionName, InputSpecs, in createAndEnsureValid()
66 const std::vector<TensorSpec> &InputSpecs, in createAndEnsureValid() argument
70 new ModelUnderTrainingRunner(Ctx, ModelPath, InputSpecs, OutputSpecs)); in createAndEnsureValid()
H A DTFUtils.cpp135 const std::vector<TensorSpec> &InputSpecs,
281 StringRef SavedModelPath, const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluatorImpl() argument
285 InputFeed(InputSpecs.size()), Input(InputSpecs.size()), in TFModelEvaluatorImpl()
301 for (size_t I = 0; I < InputSpecs.size(); ++I) { in TFModelEvaluatorImpl()
302 auto &InputSpec = InputSpecs[I]; in TFModelEvaluatorImpl()
334 StringRef SavedModelPath, const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluator() argument
337 : Impl(new TFModelEvaluatorImpl(SavedModelPath, InputSpecs, GetOutputSpecs, in TFModelEvaluator()
344 const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluator() argument
348 SavedModelPath, InputSpecs, [&](size_t I) { return OutputSpecs[I]; }, in TFModelEvaluator()
H A DInlineSizeEstimatorAnalysis.cpp222 std::vector<TensorSpec> InputSpecs{TensorSpec::createSpec<int32_t>( in InlineSizeEstimatorAnalysis() local
229 TFIR2NativeModelPath.getValue().c_str(), InputSpecs, OutputSpecs); in InlineSizeEstimatorAnalysis()
/llvm-project-15.0.7/llvm/include/llvm/Analysis/
H A DModelUnderTrainingRunner.h50 const std::vector<TensorSpec> &InputSpecs,
55 const std::vector<TensorSpec> &InputSpecs,
60 const std::vector<TensorSpec> &InputSpecs,
/llvm-project-15.0.7/llvm/unittests/Analysis/
H A DTFUtilsTest.cpp48 std::vector<TensorSpec> InputSpecs{TensorSpec::createSpec<int32_t>( in TEST() local
53 TFModelEvaluator Evaluator(getModelPath(), InputSpecs, OutputSpecs); in TEST()
89 std::vector<TensorSpec> InputSpecs{TensorSpec::createSpec<int32_t>( in TEST() local
94 TFModelEvaluator Evaluator(getModelPath(), InputSpecs, OutputSpecs); in TEST()
109 std::vector<TensorSpec> InputSpecs{ in TEST() local
116 Ctx, getModelPath(), "StatefulPartitionedCall", InputSpecs, in TEST()
/llvm-project-15.0.7/llvm/include/llvm/Analysis/Utils/
H A DTFUtils.h152 const std::vector<TensorSpec> &InputSpecs,
156 const std::vector<TensorSpec> &InputSpecs,