def METHOD_NAME(model, data_shape, out_shape):
    dtype = "float32"
    data = np.random.uniform(size=data_shape).astype(dtype)
    c2_out = get_caffe2_output(model, data, dtype)
    for target, dev in tvm.testing.enabled_targets():
        tvm_out = get_tvm_output(model, data, target, dev, out_shape, dtype)
        tvm.testing.assert_allclose(c2_out, tvm_out, rtol=1e-5, atol=1e-5)