ATLAS Offline Software
Functions
python.OnnxRuntimeInferenceConfig Namespace Reference

Functions

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

Function Documentation

◆ OnnxRuntimeInferenceToolCfg()

def 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.

10  model_fname: str = None,
11  execution_provider: Optional[OnnxRuntimeType] = None,
12  name="OnnxRuntimeInferenceTool", **kwargs):
13  """Configure OnnxRuntimeInferenceTool in Control/AthOnnx/AthOnnxComps/src"""
14 
15  acc = ComponentAccumulator()
16 
17  session_tool = acc.popToolsAndMerge(OnnxRuntimeSessionToolCfg(flags, model_fname, execution_provider))
18  kwargs["ORTSessionTool"] = session_tool
19  acc.setPrivateTools(CompFactory.AthOnnx.OnnxRuntimeInferenceTool(name, **kwargs))
20  return acc
python.JetAnalysisCommon.ComponentAccumulator
ComponentAccumulator
Definition: JetAnalysisCommon.py:302
python.OnnxRuntimeSessionConfig.OnnxRuntimeSessionToolCfg
def OnnxRuntimeSessionToolCfg(flags, str model_fname, Optional[OnnxRuntimeType] execution_provider=None, name="OnnxRuntimeSessionTool", **kwargs)
Definition: OnnxRuntimeSessionConfig.py:8
python.OnnxRuntimeInferenceConfig.OnnxRuntimeInferenceToolCfg
def OnnxRuntimeInferenceToolCfg(flags, str model_fname=None, Optional[OnnxRuntimeType] execution_provider=None, name="OnnxRuntimeInferenceTool", **kwargs)
Definition: OnnxRuntimeInferenceConfig.py:9