Searched refs:GetOutputSpecs (Results 1 – 2 of 2) sorted by relevance
136 function_ref<TensorSpec(size_t)> GetOutputSpecs,282 function_ref<TensorSpec(size_t)> GetOutputSpecs, size_t OutputSpecsSize, in TFModelEvaluatorImpl() argument324 auto OutputSpec = GetOutputSpecs(I); in TFModelEvaluatorImpl()335 function_ref<TensorSpec(size_t)> GetOutputSpecs, size_t OutputSpecsSize, in TFModelEvaluator() argument337 : Impl(new TFModelEvaluatorImpl(SavedModelPath, InputSpecs, GetOutputSpecs, in TFModelEvaluator()
157 function_ref<TensorSpec(size_t)> GetOutputSpecs,