# Copyright (c) ONNX Project Contributors # SPDX-License-Identifier: Apache-2.0 from __future__ import annotations import numpy as np from onnx.reference.ops._op import OpRunReduceNumpy class ReduceMin_1(OpRunReduceNumpy): def _run(self, data, axes=None, keepdims=None): axes = tuple(axes) if axes is not None else None if data.size == 0: maxvalue = ( np.iinfo(data.dtype).max if np.issubdtype(data.dtype, np.integer) else np.inf ) return self.reduce_constant(data, maxvalue, axes, keepdims) res = np.minimum.reduce(data, axis=axes, keepdims=keepdims == 1) if keepdims == 0 and not isinstance(res, np.ndarray): # The runtime must return a numpy array of a single float. res = np.array(res) return (res,) class ReduceMin_11(ReduceMin_1): pass class ReduceMin_18(OpRunReduceNumpy): def _run(self, data, axes=None, keepdims: int = 1, noop_with_empty_axes: int = 0): axes = self.handle_axes(axes, noop_with_empty_axes) keepdims = keepdims != 0 if data.size == 0: maxvalue = ( np.iinfo(data.dtype).max if np.issubdtype(data.dtype, np.integer) else np.inf ) return self.reduce_constant(data, maxvalue, axes, keepdims) res = np.minimum.reduce(data, axis=axes, keepdims=keepdims) if keepdims == 0 and not isinstance(res, np.ndarray): # The runtime must return a numpy array of a single float. res = np.array(res) return (res,)