def METHOD_NAME(func):
    # This TFLite graph gets lowered into
    # deptwhise_conv2d -> clip -> reshape -> max_pool -> strided_slice -> reshape
    # which gets legalized into ethosu_depthwise_conv2d -> reshape -> ehtosu_identity
    # -> ethosu_pooling -> strided_slice -> identity -> reshape -> identity
    identity3 = func.body
    reshape2 = identity3.args[0]
    identity2 = reshape2.args[0]
    strided_slice = identity2.args[0]
    max_pool = strided_slice.args[0]
    identity1 = max_pool.args[0]
    reshape1 = identity1.args[0]
    depthwise_conv2d = reshape1.args[0]
    assert identity3.op.name == "contrib.ethosu.identity"
    assert reshape2.op.name == "reshape"
    assert identity2.op.name == "contrib.ethosu.identity"
    assert strided_slice.op.name == "strided_slice"
    assert max_pool.op.name == "contrib.ethosu.pooling"
    assert identity1.op.name == "contrib.ethosu.identity"
    assert reshape1.op.name == "reshape"
    assert depthwise_conv2d.op.name == "contrib.ethosu.depthwise_conv2d"