Home
last modified time | relevance | path

Searched refs:InputSpec (Results 1 – 3 of 3) sorted by relevance

/llvm-project-15.0.7/llvm/include/llvm/Analysis/
H A DReleaseModeModelRunner.h34 ReleaseModeModelRunner(LLVMContext &Ctx, const FType &InputSpec,
37 : MLModelRunner(Ctx, MLModelRunner::Kind::Release, InputSpec.size()),
41 for (size_t I = 0; I < InputSpec.size(); ++I) {
43 CompiledModel->LookupArgIndex(FeedPrefix.str() + InputSpec[I].name());
47 setUpBufferForTensor(I, InputSpec[I], Buffer);
/llvm-project-15.0.7/llvm/lib/Analysis/
H A DTFUtils.cpp302 auto &InputSpec = InputSpecs[I]; in TFModelEvaluatorImpl() local
304 TF_GraphOperationByName(Graph.get(), (InputSpec.name()).c_str()), in TFModelEvaluatorImpl()
305 InputSpec.port()}; in TFModelEvaluatorImpl()
315 if (!checkReportAndInvalidate(InputFeed[I], InputSpec)) in TFModelEvaluatorImpl()
317 initInput(I, static_cast<TF_DataType>(getTFTypeIndex(InputSpec.type())), in TFModelEvaluatorImpl()
318 InputSpec.shape()); in TFModelEvaluatorImpl()
/llvm-project-15.0.7/llvm/unittests/Analysis/Inputs/ir2native_x86_64_model/
H A Dsaved_model.pbtxt8465 …tuple__\", \"items\": [null, 214]}}, \"input_spec\": {\"class_name\": \"InputSpec\", \"config\": {…
8507 …": null, \"bias_constraint\": null}, \"input_spec\": {\"class_name\": \"InputSpec\", \"config\": {…
8549 …": null, \"bias_constraint\": null}, \"input_spec\": {\"class_name\": \"InputSpec\", \"config\": {…