Searched refs:InputSpecs (Results 1 – 4 of 4) sorted by relevance
| /external/llvm-project/llvm/lib/Analysis/ |
| D | TFUtils.cpp | 286 const std::vector<TensorSpec> &InputSpecs,
338 StringRef SavedModelPath, const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluatorImpl() argument
342 InputFeed(InputSpecs.size()), Input(InputSpecs.size()), in TFModelEvaluatorImpl()
357 for (size_t I = 0; I < InputSpecs.size(); ++I) { in TFModelEvaluatorImpl()
358 auto &InputSpec = InputSpecs[I]; in TFModelEvaluatorImpl()
378 StringRef SavedModelPath, const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluator() argument
381 : Impl(new TFModelEvaluatorImpl(SavedModelPath, InputSpecs, GetOutputSpecs, in TFModelEvaluator()
388 const std::vector<TensorSpec> &InputSpecs, in TFModelEvaluator() argument
392 SavedModelPath, InputSpecs, [&](size_t I) { return OutputSpecs[I]; }, in TFModelEvaluator()
|
| D | InlineSizeEstimatorAnalysis.cpp | 224 std::vector<TensorSpec> InputSpecs{TensorSpec::createSpec<int32_t>( in InlineSizeEstimatorAnalysis() local
231 TFIR2NativeModelPath.getValue().c_str(), InputSpecs, OutputSpecs); in InlineSizeEstimatorAnalysis()
|
| /external/llvm-project/llvm/unittests/Analysis/ |
| D | TFUtilsTest.cpp | 43 std::vector<TensorSpec> InputSpecs{TensorSpec::createSpec<int32_t>( in TEST() local
48 TFModelEvaluator Evaluator(getModelPath(), InputSpecs, OutputSpecs); in TEST()
84 std::vector<TensorSpec> InputSpecs{TensorSpec::createSpec<int32_t>( in TEST() local
89 TFModelEvaluator Evaluator(getModelPath(), InputSpecs, OutputSpecs); in TEST()
|
| /external/llvm-project/llvm/include/llvm/Analysis/Utils/ |
| D | TFUtils.h | 210 const std::vector<TensorSpec> &InputSpecs,
214 const std::vector<TensorSpec> &InputSpecs,
|