ATLAS Offline Software
Functions | Variables
AthExOnnxRuntime_test_infer Namespace Reference

Functions

def AthExOnnxRuntimeExampleCfg (flags, name="AthOnnxExample", **kwargs)
 

Variables

 flags = initConfigFlags()
 
 ExecutionProvider
 
 acc = MainServicesCfg(flags)
 
 withDetails
 
 True
 
 summariseProps
 

Function Documentation

◆ AthExOnnxRuntimeExampleCfg()

def AthExOnnxRuntime_test_infer.AthExOnnxRuntimeExampleCfg (   flags,
  name = "AthOnnxExample",
**  kwargs 
)

Definition at line 9 of file AthExOnnxRuntime_test_infer.py.

9 def AthExOnnxRuntimeExampleCfg(flags, name="AthOnnxExample", **kwargs):
10  acc = ComponentAccumulator()
11 
12  model_fname = "dev/MLTest/2020-03-02/MNIST_testModel.onnx"
13  execution_provider = OnnxRuntimeType.CPU
14  from AthOnnxComps.OnnxRuntimeInferenceConfig import OnnxRuntimeInferenceToolCfg
15  kwargs.setdefault("ORTInferenceTool", acc.popToolsAndMerge(
16  OnnxRuntimeInferenceToolCfg(flags, model_fname, execution_provider)
17  ))
18 
19  input_data = "dev/MLTest/2020-03-31/t10k-images-idx3-ubyte"
20  kwargs.setdefault("BatchSize", 3)
21  kwargs.setdefault("InputDataPixel", input_data)
22  kwargs.setdefault("OutputLevel", Constants.DEBUG)
23  acc.addEventAlgo(CompFactory.AthOnnx.EvaluateModelWithAthInfer(name, **kwargs))
24 
25  return acc
26 

Variable Documentation

◆ acc

AthExOnnxRuntime_test_infer.acc = MainServicesCfg(flags)

Definition at line 38 of file AthExOnnxRuntime_test_infer.py.

◆ ExecutionProvider

AthExOnnxRuntime_test_infer.ExecutionProvider

Definition at line 35 of file AthExOnnxRuntime_test_infer.py.

◆ flags

AthExOnnxRuntime_test_infer.flags = initConfigFlags()

Definition at line 34 of file AthExOnnxRuntime_test_infer.py.

◆ summariseProps

AthExOnnxRuntime_test_infer.summariseProps

Definition at line 40 of file AthExOnnxRuntime_test_infer.py.

◆ True

AthExOnnxRuntime_test_infer.True

Definition at line 40 of file AthExOnnxRuntime_test_infer.py.

◆ withDetails

AthExOnnxRuntime_test_infer.withDetails

Definition at line 40 of file AthExOnnxRuntime_test_infer.py.

python.JetAnalysisCommon.ComponentAccumulator
ComponentAccumulator
Definition: JetAnalysisCommon.py:302
python.OnnxRuntimeInferenceConfig.OnnxRuntimeInferenceToolCfg
def OnnxRuntimeInferenceToolCfg(flags, str model_fname=None, Optional[OnnxRuntimeType] execution_provider=None, name="OnnxRuntimeInferenceTool", **kwargs)
Definition: OnnxRuntimeInferenceConfig.py:9
AthExOnnxRuntime_test_infer.AthExOnnxRuntimeExampleCfg
def AthExOnnxRuntimeExampleCfg(flags, name="AthOnnxExample", **kwargs)
Definition: AthExOnnxRuntime_test_infer.py:9