test.py 250 Bytes
Newer Older
Paul's avatar
Paul committed
1
import migraphx
Paul's avatar
Paul committed
2
3

p = migraphx.parse_onnx("conv_relu_maxpool.onnx")
Paul's avatar
Paul committed
4
5
6
7
8
9
10
11
p.compile(migraphx.get_target("cpu"))

params = {}
for key, value in p.get_parameter_shapes().items():
    params[key] = migraphx.generate_argument(value)

r = p.run(params)
print(r)