Skip to content

Commit 18eeda8

Browse files
committed
Add gte/lte ops; Remove numeric constraint on Not
1 parent a9bbd16 commit 18eeda8

File tree

2 files changed

+103
-1
lines changed

2 files changed

+103
-1
lines changed

backends/src/main/scala/ORTOperatorBackendAtoL213.scala

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ class ORTOperatorBackendAtoL
6161
// with GlobalLpPool
6262
with GlobalMaxPool
6363
with Greater
64+
with GreaterOrEqual
6465
with HardSigmoid
6566
with Hardmax
6667
with Identity
@@ -74,6 +75,7 @@ class ORTOperatorBackendAtoL
7475
// with LabelEncoder
7576
with LeakyRelu
7677
with Less
78+
with LessOrEqual
7779
// with LinearClassifier
7880
// with LinearRegressor
7981
with Log

core/src/main/scala/ONNX213.scala

Lines changed: 101 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5045,6 +5045,57 @@ package object onnx {
50455045
](name, "GlobalMaxPool", allInputs, map))
50465046
}
50475047
}
5048+
5049+
trait GreaterOrEqual extends Operator {
5050+
def GreaterOrEqual12[
5051+
@sp T : Numeric: ClassTag,
5052+
@sp T1 : ClassTag
5053+
](name: String, A: Option[Tensor[T]], B: Option[Tensor[T]])
5054+
(implicit evT: Contains[
5055+
T,
5056+
Union[Float16]#or[Float]#or[Double]#or[UByte]#or[UShort]#or[UInt]#or[ULong]#or[Byte]#or[
5057+
Short
5058+
]#or[Int]#or[Long]#or[UNil]#create
5059+
],
5060+
evT1: Contains[T1, Union[Boolean]#or[UNil]#create]
5061+
): (Tensor[T1]) = {
5062+
val map: Map[String, Any] = Map()
5063+
val allInputs = (
5064+
A,
5065+
B,
5066+
None: Option[Any],
5067+
None: Option[Any],
5068+
None: Option[Any],
5069+
None: Option[Any],
5070+
None: Option[Any],
5071+
None: Option[Any],
5072+
None: Option[Any]
5073+
)
5074+
(callOp[
5075+
Option[Tensor[T]],
5076+
Option[Tensor[T]],
5077+
Any,
5078+
Any,
5079+
Any,
5080+
Any,
5081+
Any,
5082+
Any,
5083+
Any,
5084+
Tensor[T1],
5085+
Any,
5086+
Any,
5087+
Any,
5088+
Any,
5089+
Any,
5090+
Any,
5091+
Any,
5092+
Any
5093+
](name, "GreaterOrEqual", allInputs, map))
5094+
}
5095+
}
5096+
5097+
5098+
50485099
trait Greater extends Operator {
50495100

50505101
def Greater1[@sp T: Numeric: ClassTag, @sp T1: Numeric: ClassTag](
@@ -6110,6 +6161,55 @@ package object onnx {
61106161
](name, "LeakyRelu", allInputs, map))
61116162
}
61126163
}
6164+
6165+
trait LessOrEqual extends Operator {
6166+
def LessOrEqual12[
6167+
@sp T : Numeric: ClassTag,
6168+
@sp T1 : ClassTag
6169+
](name: String, A: Option[Tensor[T]], B: Option[Tensor[T]])
6170+
(implicit evT: Contains[
6171+
T,
6172+
Union[Float16]#or[Float]#or[Double]#or[UByte]#or[UShort]#or[UInt]#or[ULong]#or[Byte]#or[
6173+
Short
6174+
]#or[Int]#or[Long]#or[UNil]#create
6175+
],
6176+
evT1: Contains[T1, Union[Boolean]#or[UNil]#create]
6177+
): (Tensor[T1]) = {
6178+
val map: Map[String, Any] = Map()
6179+
val allInputs = (
6180+
A,
6181+
B,
6182+
None: Option[Any],
6183+
None: Option[Any],
6184+
None: Option[Any],
6185+
None: Option[Any],
6186+
None: Option[Any],
6187+
None: Option[Any],
6188+
None: Option[Any]
6189+
)
6190+
(callOp[
6191+
Option[Tensor[T]],
6192+
Option[Tensor[T]],
6193+
Any,
6194+
Any,
6195+
Any,
6196+
Any,
6197+
Any,
6198+
Any,
6199+
Any,
6200+
Tensor[T1],
6201+
Any,
6202+
Any,
6203+
Any,
6204+
Any,
6205+
Any,
6206+
Any,
6207+
Any,
6208+
Any
6209+
](name, "LessOrEqual", allInputs, map))
6210+
}
6211+
}
6212+
61136213
trait Less extends Operator {
61146214

61156215
def Less1[@sp T: Numeric: ClassTag, @sp T1: Numeric: ClassTag](
@@ -8305,7 +8405,7 @@ package object onnx {
83058405
}
83068406
trait Not extends Operator {
83078407

8308-
def Not1[@sp T: Numeric: ClassTag](name: String, X: Option[Tensor[T]])(implicit
8408+
def Not1[@sp T: ClassTag](name: String, X: Option[Tensor[T]])(implicit
83098409
evT: Contains[T, Union[Boolean]#or[UNil]#create]
83108410
): (Tensor[T]) = {
83118411
val map: Map[String, Any] = Map()

0 commit comments

Comments
 (0)