def METHOD_NAME(
    self,
    floating_point_func: Callable[[np.ndarray], np.ndarray],
    input_arg=None,
    in_scale=relay.const(1.0, dtype="float32"),
    in_zero_point=relay.const(0, dtype="int32"),
    out_scale=relay.const(1.0, dtype="float32"),
    out_zero_point=relay.const(0, dtype="int32"),
    in_axis=-1,
    out_axis=-1,
    in_dtype="uint8",
    out_dtype="uint8",