I am done

This commit is contained in:
2024-10-30 22:14:35 +01:00
parent 720dc28c09
commit 40e2a747cf
36901 changed files with 5011519 additions and 0 deletions

View File

@ -0,0 +1,16 @@
# Copyright (c) ONNX Project Contributors
# SPDX-License-Identifier: Apache-2.0
from __future__ import annotations
import numpy as np
from onnx.reference.op_run import OpRun
class Trilu(OpRun):
def _run(self, x, k=None, upper=None): # type: ignore
k = 0 if k is None else int(k)
if upper: # type: ignore
return (np.triu(x, k),)
return (np.tril(x, k).astype(x.dtype),)