Searched refs:OutputSpecs (Results 1 – 5 of 5) sorted by relevance
| /freebsd/contrib/llvm-project/llvm/lib/Analysis/ |
| H A D | ModelUnderTrainingRunner.cpp | 96 const std::vector<TensorSpec> &OutputSpecs, in ModelUnderTrainingRunner() argument
99 OutputSpecs(OutputSpecs), ExtraOutputsForLogging(ExtraOutputsForLogging) { in ModelUnderTrainingRunner()
101 std::make_unique<TFModelEvaluator>(ModelPath, InputSpecs, OutputSpecs); in ModelUnderTrainingRunner()
130 std::vector<TensorSpec> OutputSpecs; in createAndEnsureValid() local
132 append_range(OutputSpecs, in createAndEnsureValid()
146 Ctx, ModelPath, InputSpecs, OutputSpecs, ExtraOutputsForLogging)); in createAndEnsureValid()
|
| H A D | TFLiteUtils.cpp | 57 const std::vector<TensorSpec> &OutputSpecs,
102 const std::vector<TensorSpec> &OutputSpecs, const char *Tags = "serve") in TFModelEvaluatorImpl() argument
103 : Input(InputSpecs.size()), Output(OutputSpecs.size()) { in TFModelEvaluatorImpl()
174 for (size_t I = 0; I < OutputSpecs.size(); ++I) { in TFModelEvaluatorImpl()
175 const auto &OutputSpec = OutputSpecs[I]; in TFModelEvaluatorImpl()
186 const std::vector<TensorSpec> &OutputSpecs, in TFModelEvaluator() argument
188 : Impl(new TFModelEvaluatorImpl(SavedModelPath, InputSpecs, OutputSpecs, in TFModelEvaluator()
|
| H A D | InlineSizeEstimatorAnalysis.cpp | 225 std::vector<TensorSpec> OutputSpecs{ 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 | 63 const std::vector<TensorSpec> &OutputSpecs,
70 const std::vector<TensorSpec> OutputSpecs;
69 const std::vector<TensorSpec> OutputSpecs; global() variable
|
| /freebsd/contrib/llvm-project/llvm/include/llvm/Analysis/Utils/ |
| H A D | TFUtils.h | 78 const std::vector<TensorSpec> &OutputSpecs,
|