14 lines
291 B
Python
14 lines
291 B
Python
# 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 OpRunBinaryComparison
|
|
|
|
|
|
class Equal(OpRunBinaryComparison):
|
|
def _run(self, a, b):
|
|
return (np.equal(a, b),)
|