def METHOD_NAME(self):
    code = """
        <opset_import: [ "" : 18 ], domain: "local">
        f (y, z) => (w) {
            x = Add(y, z)
            w = Mul(x, y)
        }
    """
    self._check(code, [float_type_, float_type_], [], [float_type_])
    self._check(code, [int32_type_, int32_type_], [], [int32_type_])
    self._check_fails(code, [float_type_, int32_type_], [])