Coverage for mlprodict/onnxrt/ops_cpu/op_isinf.py: 100%
18 statements
« prev ^ index » next coverage.py v7.1.0, created at 2023-02-04 02:28 +0100
« prev ^ index » next coverage.py v7.1.0, created at 2023-02-04 02:28 +0100
1# -*- encoding: utf-8 -*-
2# pylint: disable=E0203,E1101,C0111
3"""
4@file
5@brief Runtime operator.
6"""
7import numpy
8from ._op import OpRunUnary
11class IsInf(OpRunUnary):
13 atts = {'detect_negative': 1, 'detect_positive': 1}
15 def __init__(self, onnx_node, desc=None, **options):
16 OpRunUnary.__init__(self, onnx_node, desc=desc,
17 expected_attributes=IsInf.atts,
18 **options)
20 def _run(self, data, attributes=None, verbose=0, fLOG=None): # pylint: disable=W0221
21 if self.detect_negative:
22 if self.detect_positive:
23 return (numpy.isinf(data), )
24 return (numpy.isneginf(data), )
25 if self.detect_positive:
26 return (numpy.isposinf(data), )
27 res = numpy.full(data.shape, dtype=numpy.bool_, fill_value=False)
28 return (res, )
30 def to_python(self, inputs):
31 return self._to_python_numpy(inputs, 'isnan')