Searched refs:InputSpecs (Results 1 – 7 of 7) sorted by relevance
| /freebsd/contrib/llvm-project/llvm/lib/Analysis/ |
| H A D | ModelUnderTrainingRunner.cpp | 95 const std::vector<TensorSpec> &InputSpecs, in ModelUnderTrainingRunner() argument
98 : MLModelRunner(Ctx, MLModelRunner::Kind::Development, InputSpecs.size()), in ModelUnderTrainingRunner()
101 std::make_unique<TFModelEvaluator>(ModelPath, InputSpecs, OutputSpecs); in ModelUnderTrainingRunner()
108 for (size_t I = 0, E = InputSpecs.size(); I < E; ++I) { in ModelUnderTrainingRunner()
109 setUpBufferForTensor(I, InputSpecs[I], Evaluator->getUntypedInput(I)); in ModelUnderTrainingRunner()
125 const std::vector<TensorSpec> &InputSpecs, in createAndEnsureValid() argument
146 Ctx, ModelPath, InputSpecs, OutputSpecs, ExtraOutputsForLogging)); in createAndEnsureValid()
|
| H A D | TFLiteUtils.cpp | 56 const std::vector<TensorSpec> &InputSpecs,
101 StringRef SavedModelPath, const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluatorImpl() argument
103 : Input(InputSpecs.size()), Output(OutputSpecs.size()) { in TFModelEvaluatorImpl()
151 for (size_t I = 0; I < InputSpecs.size(); ++I) { in TFModelEvaluatorImpl()
152 auto &InputSpec = InputSpecs[I]; in TFModelEvaluatorImpl()
163 InputSpecs[I].getTotalTensorBufferSize()); in TFModelEvaluatorImpl()
185 const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluator() argument
188 : Impl(new TFModelEvaluatorImpl(SavedModelPath, InputSpecs, OutputSpecs, in TFModelEvaluator()
|
| H A D | InteractiveModelRunner.cpp | 30 InputSpecs(Inputs), OutputSpec(Advice), in InteractiveModelRunner()
43 Log = std::make_unique<Logger>(std::move(OutStream), InputSpecs, Advice, in InteractiveModelRunner()
48 for (size_t I = 0; I < InputSpecs.size(); ++I) in InteractiveModelRunner()
49 setUpBufferForTensor(I, InputSpecs[I], nullptr); in InteractiveModelRunner()
60 for (size_t I = 0; I < InputSpecs.size(); ++I) in evaluateUntyped()
|
| H A D | InlineSizeEstimatorAnalysis.cpp | 221 std::vector<TensorSpec> InputSpecs{TensorSpec::createSpec<int32_t>( in InlineSizeEstimatorAnalysis() local
228 TFIR2NativeModelPath.getValue().c_str(), InputSpecs, OutputSpecs); in InlineSizeEstimatorAnalysis()
|
| /freebsd/contrib/llvm-project/llvm/include/llvm/Analysis/ |
| H A D | ModelUnderTrainingRunner.h | 57 const std::vector<TensorSpec> &InputSpecs,
62 const std::vector<TensorSpec> &InputSpecs,
|
| H A D | InteractiveModelRunner.h | 63 const std::vector<TensorSpec> InputSpecs; variable
|
| /freebsd/contrib/llvm-project/llvm/include/llvm/Analysis/Utils/ |
| H A D | TFUtils.h | 77 const std::vector<TensorSpec> &InputSpecs,
|