()
| 79 | |
| 80 | |
| 81 | def test_nonzero(): |
| 82 | p = migraphx.parse_onnx("nonzero_dynamic_test.onnx") |
| 83 | print(p) |
| 84 | print("Compiling ...") |
| 85 | p.compile(migraphx.get_target("gpu")) |
| 86 | print(p) |
| 87 | params = {} |
| 88 | |
| 89 | shapes = p.get_parameter_shapes() |
| 90 | params["data"] = migraphx.create_argument( |
| 91 | migraphx.shape(type='bool_type', lens=shapes["data"].lens()), |
| 92 | [1, 1, 0, 1]) |
| 93 | |
| 94 | r = p.run(params) |
| 95 | print(r) |
| 96 | |
| 97 | |
| 98 | def test_fp16_imagescaler(): |
no test coverage detected