Function _get_max_identity

imperative/python/megengine/xla/rules/reduction.py:24–32

(dtype)

def _get_max_identity(dtype) -> np.ndarray:
    if dtype == np.float32 or dtype == np.float64 or dtype == np.float16:
        return np.array(-np.inf, dtype)
    elif (
        dtype == np.int32 or dtype == np.int64 or dtype == np.int16 or dtype == np.int8
    ):
        return np.array(np.iinfo(dtype).min, dtype)
    else:
        assert False, f"unsupported dtype for max: {dtype}"


def _get_min_identity(dtype) -> np.ndarray: