Searched refs:OutputSpecs (Results 1 – 6 of 6) sorted by relevance
| /llvm-project-15.0.7/llvm/lib/Analysis/ |
| H A D | ModelUnderTrainingRunner.cpp | 24 const std::vector<LoggedFeatureSpec> &OutputSpecs) in ModelUnderTrainingRunner() argument
26 OutputSpecs(OutputSpecs) { in ModelUnderTrainingRunner()
28 ModelPath, InputSpecs, [&](size_t I) { return OutputSpecs[I].Spec; }, in ModelUnderTrainingRunner()
29 OutputSpecs.size()); in ModelUnderTrainingRunner()
67 const std::vector<LoggedFeatureSpec> &OutputSpecs) { in createAndEnsureValid() argument
70 new ModelUnderTrainingRunner(Ctx, ModelPath, InputSpecs, OutputSpecs)); in createAndEnsureValid()
|
| H A D | InlineSizeEstimatorAnalysis.cpp | 226 std::vector<TensorSpec> OutputSpecs{ in InlineSizeEstimatorAnalysis() local
229 TFIR2NativeModelPath.getValue().c_str(), InputSpecs, OutputSpecs); in InlineSizeEstimatorAnalysis()
|
| H A D | TFUtils.cpp | 345 const std::vector<TensorSpec> &OutputSpecs, in TFModelEvaluator() argument
348 SavedModelPath, InputSpecs, [&](size_t I) { return OutputSpecs[I]; }, in TFModelEvaluator()
349 OutputSpecs.size(), Tags) {}
|
| /llvm-project-15.0.7/llvm/include/llvm/Analysis/ |
| H A D | ModelUnderTrainingRunner.h | 36 return OutputSpecs; in outputLoggedFeatureSpecs()
56 const std::vector<LoggedFeatureSpec> &OutputSpecs);
61 const std::vector<LoggedFeatureSpec> &OutputSpecs);
64 const std::vector<LoggedFeatureSpec> OutputSpecs; variable
|
| /llvm-project-15.0.7/llvm/unittests/Analysis/ |
| H A D | TFUtilsTest.cpp | 50 std::vector<TensorSpec> OutputSpecs{ in TEST() local
53 TFModelEvaluator Evaluator(getModelPath(), InputSpecs, OutputSpecs); in TEST()
91 std::vector<TensorSpec> OutputSpecs{ in TEST() local
94 TFModelEvaluator Evaluator(getModelPath(), InputSpecs, OutputSpecs); in TEST()
|
| /llvm-project-15.0.7/llvm/include/llvm/Analysis/Utils/ |
| H A D | TFUtils.h | 153 const std::vector<TensorSpec> &OutputSpecs,
|