Function test_conv2d

tests/python/relax/test_codegen_coreml.py:263–274

()

@requires_coreml_runtime
def test_conv2d():
    x = relax.Var("x", relax.TensorStructInfo([1, 3, 224, 224], "float32"))
    w = relax.const(np.zeros((16, 3, 3, 3), dtype="float32"))
    bb = relax.BlockBuilder()
    with bb.function("main", [x]):
        with bb.dataflow():
            lv0 = bb.emit(relax.op.nn.conv2d(x, w, strides=[2, 2], padding=[1, 1, 1, 1]))
            gv = bb.emit_output(lv0)
        bb.emit_func_output(gv)
    mod = bb.get()
    x_data = tvm.runtime.tensor(np.random.rand(1, 3, 224, 224).astype("float32"), dev)
    verify(mod, [x_data])


@requires_coreml_runtime