Loading [MathJax]/extensions/tex2jax.js
ATLAS Offline Software
Main Page
Related Pages
Modules
Namespaces
Namespace List
Namespace Members
All
_
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
x
y
z
Functions
_
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
x
y
z
Variables
_
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
x
y
z
Typedefs
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
x
z
Enumerations
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
z
Enumerator
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
x
y
z
Classes
Class List
Class Index
Class Hierarchy
Class Members
All
:
_
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
x
y
z
~
Functions
_
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
x
y
z
~
Variables
_
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
x
y
z
Typedefs
_
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
x
y
z
Enumerations
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
x
y
z
Enumerator
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
x
y
z
Properties
Related Functions
:
a
b
c
d
e
f
g
h
i
j
l
m
n
o
p
r
s
t
v
w
x
z
Files
File List
File Members
All
$
_
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
x
y
z
Functions
_
a
b
c
d
e
f
g
h
i
j
l
m
n
o
p
q
r
s
t
u
v
w
x
z
Variables
$
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
x
y
z
Typedefs
a
b
c
d
e
f
g
h
i
j
l
m
n
o
p
q
r
s
t
u
v
w
x
z
Enumerations
a
b
c
d
e
f
g
h
i
l
m
n
o
p
r
s
t
v
x
z
Enumerator
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
x
y
z
Macros
_
a
b
c
d
e
f
g
h
i
j
k
l
m
n
o
p
q
r
s
t
u
v
w
x
y
z
GitLab
LXR
•
All
Classes
Namespaces
Files
Functions
Variables
Typedefs
Enumerations
Enumerator
Properties
Friends
Macros
Modules
Pages
Control
AthOnnx
AthOnnxComps
python
OnnxRuntimeFlags.py
Go to the documentation of this file.
1
# Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
2
3
from
AthenaConfiguration.AthConfigFlags
import
AthConfigFlags
4
from
AthenaConfiguration.Enums
import
FlagEnum
5
6
class
OnnxRuntimeType
(FlagEnum):
7
CPU =
'CPU'
8
CUDA =
'CUDA'
9
# possible future backends. Uncomment when implemented.
10
# DML = 'DML'
11
# DNNL = 'DNNL'
12
# NUPHAR = 'NUPHAR'
13
# OPENVINO = 'OPENVINO'
14
# ROCM = 'ROCM'
15
# TENSORRT = 'TENSORRT'
16
# VITISAI = 'VITISAI'
17
# VULKAN = 'VULKAN'
18
19
def
createOnnxRuntimeFlags
():
20
icf = AthConfigFlags()
21
22
icf.addFlag(
"AthOnnx.ExecutionProvider"
, OnnxRuntimeType.CPU, type=OnnxRuntimeType)
23
24
return
icf
25
26
if
__name__ ==
"__main__"
:
27
28
flags =
createOnnxRuntimeFlags
()
29
flags.dump()
python.OnnxRuntimeFlags.createOnnxRuntimeFlags
def createOnnxRuntimeFlags()
Definition:
OnnxRuntimeFlags.py:19
python.OnnxRuntimeFlags.OnnxRuntimeType
Definition:
OnnxRuntimeFlags.py:6
Generated on Mon Apr 21 2025 21:16:18 for ATLAS Offline Software by
1.8.18
1.8.18