python.OnnxRuntimeSessionConfig Namespace Reference

Functions
def	OnnxRuntimeSessionToolCfg (flags, str model_fname, Optional[OnnxRuntimeType] execution_provider=None, name="OnnxRuntimeSessionTool", **kwargs)

Function Documentation

OnnxRuntimeSessionToolCfg()

def python.OnnxRuntimeSessionConfig.OnnxRuntimeSessionToolCfg	(		flags,
		str	model_fname,
		Optional[OnnxRuntimeType]	execution_provider = None,
			name = "OnnxRuntimeSessionTool",
		**	kwargs
	)

"Configure OnnxRuntimeSessionTool in Control/AthOnnx/AthOnnxComps/src

Definition at line 8 of file OnnxRuntimeSessionConfig.py.

def OnnxRuntimeSessionToolCfg(flags,
                              model_fname: str,
                              execution_provider: Optional[OnnxRuntimeType] = None, 
                              name="OnnxRuntimeSessionTool", **kwargs):
    """"Configure OnnxRuntimeSessionTool in Control/AthOnnx/AthOnnxComps/src"""
    
    acc = ComponentAccumulator()
 
    
    execution_provider = flags.AthOnnx.ExecutionProvider if execution_provider is None else execution_provider
    name += execution_provider.name
 
    if "OnnxRuntimeSvc" not in kwargs:
        from AthOnnxComps.OnnxRuntimeSvcConfig import OnnxRuntimeSvcCfg
        kwargs.setdefault("OnnxRuntimeSvc", acc.getPrimaryAndMerge(OnnxRuntimeSvcCfg(flags)))
    kwargs.setdefault("ModelFileName", model_fname)
    if execution_provider is OnnxRuntimeType.CPU:
        acc.setPrivateTools(CompFactory.AthOnnx.OnnxRuntimeSessionToolCPU(name, **kwargs))
    elif execution_provider is OnnxRuntimeType.CUDA:
        acc.setPrivateTools(CompFactory.AthOnnx.OnnxRuntimeSessionToolCUDA(name, **kwargs))
    else:
        raise ValueError("Unknown OnnxRuntime Execution Provider: %s" % execution_provider)
 
    return acc