ATLAS Offline Software
Loading...
Searching...
No Matches
python.OnnxRuntimeInferenceConfig Namespace Reference

Functions

 OnnxRuntimeInferenceToolCfg (flags, str model_fname=None, Optional[OnnxRuntimeType] execution_provider=None, name="OnnxRuntimeInferenceTool", **kwargs)

Function Documentation

◆ OnnxRuntimeInferenceToolCfg()

python.OnnxRuntimeInferenceConfig.OnnxRuntimeInferenceToolCfg ( flags,
str model_fname = None,
Optional[OnnxRuntimeType] execution_provider = None,
name = "OnnxRuntimeInferenceTool",
** kwargs )
Configure OnnxRuntimeInferenceTool in Control/AthOnnx/AthOnnxComps/src

Definition at line 9 of file OnnxRuntimeInferenceConfig.py.

12 name="OnnxRuntimeInferenceTool", **kwargs):
13 """Configure OnnxRuntimeInferenceTool in Control/AthOnnx/AthOnnxComps/src"""
14
15 acc = ComponentAccumulator()
16
17 if "OnnxRuntimeSvc" not in kwargs:
18 from AthOnnxComps.OnnxRuntimeSvcConfig import OnnxRuntimeSvcCfg
19 kwargs.setdefault("OnnxRuntimeSvc", acc.getPrimaryAndMerge(OnnxRuntimeSvcCfg(flags)))
20 kwargs.setdefault("ORTSessionTool", acc.popToolsAndMerge(OnnxRuntimeSessionToolCfg(flags, model_fname, execution_provider)))
21 acc.setPrivateTools(CompFactory.AthOnnx.OnnxRuntimeInferenceTool(name, **kwargs))
22 return acc