()
| 77 | def test_tensorrt_offload(): |
| 78 | @memoize("relax.tests.test_codegen_tensorrt.conv2d_residual") |
| 79 | def get_ref(): |
| 80 | data_np = np.random.randn(1, 64, 56, 56).astype("float32") |
| 81 | weight1_np = np.random.randn(64, 64, 3, 3).astype("float32") |
| 82 | weight2_np = np.random.randn(64, 64, 3, 3).astype("float32") |
| 83 | inputs = [data_np, weight1_np, weight2_np] |
| 84 | ref = build_and_run(Conv2dResidualBlock, inputs, "llvm", legalize=True) |
| 85 | return inputs, ref |
| 86 | |
| 87 | inputs, ref = get_ref() |
| 88 |
no test coverage detected
searching dependent graphs…