Skip to content

Commit fb2bf99

Browse files
EmergentOrderEmergentOrder
committed
Fix backend compilation
1 parent 2a84b67 commit fb2bf99

File tree

1 file changed

+147
-19
lines changed

1 file changed

+147
-19
lines changed

‎backends/src/main/scala/NGraphBackend.scala‎

Lines changed: 147 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -162,6 +162,17 @@ class NGraphBackend(onnxBytes: Array[Byte])
162162
???
163163
}
164164

165+
def ArgMax11[@sp T: Numeric: ClassTag](
166+
name: String,
167+
axis: Option[(Int)] = None,
168+
keepdims: Option[(Int)] = None,
169+
data: Option[Tensor[T]]
170+
)(
171+
implicit evT: Contains[T, Union[UByte]#or[UShort]#or[UInt]#or[ULong]#or[Byte]#or[Short]#or[
172+
Int
173+
]#or[Long]#or[Float16]#or[Float]#or[Double]#or[UNil]#create]
174+
): (Tensor[Long]) = ???
175+
165176
def ArgMin1[@sp T: Numeric: ClassTag](
166177
name: String,
167178
axis: Option[(Int)] = None,
@@ -177,6 +188,17 @@ class NGraphBackend(onnxBytes: Array[Byte])
177188
???
178189
}
179190

191+
def ArgMin11[@sp T: Numeric: ClassTag](
192+
name: String,
193+
axis: Option[(Int)] = None,
194+
keepdims: Option[(Int)] = None,
195+
data: Option[Tensor[T]]
196+
)(
197+
implicit evT: Contains[T, Union[UByte]#or[UShort]#or[UInt]#or[ULong]#or[Byte]#or[Short]#or[
198+
Int
199+
]#or[Long]#or[Float16]#or[Float]#or[Double]#or[UNil]#create]
200+
): (Tensor[Long]) = ???
201+
180202
def Constant1[@sp T: Numeric: ClassTag](
181203
name: String,
182204
value: Option[(Tensor[T])]
@@ -203,28 +225,57 @@ class NGraphBackend(onnxBytes: Array[Byte])
203225
(trinaryOpNoAttrs(name, "Constant", value, None, None))
204226
}
205227

206-
def Equal1[@sp T: Numeric: ClassTag, @sp T1: Numeric: ClassTag](
207-
name: String,
208-
axis: Option[(Int)],
209-
broadcast: Option[(Int)],
210-
A: Option[Tensor[T]],
211-
B: Option[Tensor[T]]
212-
)(
213-
implicit evT: Contains[T, Union[Boolean]#or[Int]#or[Long]#or[UNil]#create],
214-
evT1: Contains[T1, Union[Boolean]#or[UNil]#create]
215-
): (Tensor[T1]) = ???
216-
217-
def Equal7[@sp T: Numeric: ClassTag, @sp T1: Numeric: ClassTag](
218-
name: String,
219-
A: Option[Tensor[T]],
220-
B: Option[Tensor[T]]
221-
)(
222-
implicit evT: Contains[T, Union[Boolean]#or[Int]#or[Long]#or[UNil]#create],
223-
evT1: Contains[T1, Union[Boolean]#or[UNil]#create]
224-
): (Tensor[T1]) = {
228+
def Constant11[@sp T: Numeric: ClassTag](
229+
name: String,
230+
sparse_value: Option[(SparseTensor[T])] = None,
231+
value: Option[(Tensor[T])] = None
232+
)(
233+
implicit evT: Contains[
234+
T,
235+
Union[Float16]#or[Float]#or[Double]#or[UByte]#or[UShort]#or[UInt]#or[ULong]#or[Byte]#or[
236+
Short
237+
]#or[Int]#or[Long]#or[String]#or[Boolean]#or[Complex[Float]]#or[Complex[Double]]#or[UNil]#create
238+
]
239+
): (Tensor[T]) = ???
240+
241+
242+
def Equal1[@sp T: Numeric: ClassTag, @sp T1: Numeric: ClassTag](
243+
name: String,
244+
axis: Option[(Int)] = None,
245+
broadcast: Option[(Int)] = None,
246+
A: Option[Tensor[T]],
247+
B: Option[Tensor[T]]
248+
)(
249+
implicit evT: Contains[T, Union[Boolean]#or[Int]#or[Long]#or[UByte]#or[UShort]#or[UInt]#or[
250+
ULong
251+
]#or[Byte]#or[Short]#or[Float16]#or[Float]#or[Double]#or[UNil]#create],
252+
evT1: Contains[T1, Union[Boolean]#or[UNil]#create]
253+
): (Tensor[T1]) = ???
254+
255+
def Equal7[@sp T: Numeric: ClassTag, @sp T1: Numeric: ClassTag](
256+
name: String,
257+
A: Option[Tensor[T]],
258+
B: Option[Tensor[T]]
259+
)(
260+
implicit evT: Contains[T, Union[Boolean]#or[Int]#or[Long]#or[UByte]#or[UShort]#or[UInt]#or[
261+
ULong
262+
]#or[Byte]#or[Short]#or[Float16]#or[Float]#or[Double]#or[UNil]#create],
263+
evT1: Contains[T1, Union[Boolean]#or[UNil]#create]
264+
): (Tensor[T1]) = {
225265
//(trinaryOpNoAttrs(name, "Equal", A, B, None))
226266
???
227267
}
268+
269+
def Equal11[@sp T: Numeric: ClassTag, @sp T1: Numeric: ClassTag](
270+
name: String,
271+
A: Option[Tensor[T]],
272+
B: Option[Tensor[T]]
273+
)(
274+
implicit evT: Contains[T, Union[Boolean]#or[Int]#or[Long]#or[UByte]#or[UShort]#or[UInt]#or[
275+
ULong
276+
]#or[Byte]#or[Short]#or[Float16]#or[Float]#or[Double]#or[UNil]#create],
277+
evT1: Contains[T1, Union[Boolean]#or[UNil]#create]
278+
): (Tensor[T1]) = ???
228279

229280
def GlobalAveragePool1[@sp T: Numeric: ClassTag](
230281
name: String,
@@ -294,6 +345,13 @@ class NGraphBackend(onnxBytes: Array[Byte])
294345
(trinaryOp(name, "Softmax", input, None, None, map))
295346
}
296347

348+
def Softmax11[@sp T: Numeric: ClassTag](
349+
name: String,
350+
axis: Option[(Int)] = None,
351+
input: Option[Tensor[T]]
352+
)(implicit evT: Contains[T, Union[Float16]#or[Float]#or[Double]#or[UNil]#create]): (Tensor[T]) = ???
353+
354+
297355
def Mul1[@sp T: Numeric: ClassTag](
298356
name: String,
299357
axis: Option[(Int)] = None,
@@ -421,6 +479,22 @@ class NGraphBackend(onnxBytes: Array[Byte])
421479
(trinaryOp(name, "Gemm", A, B, C, map))
422480
}
423481

482+
def Gemm11[@sp T: Numeric: ClassTag](
483+
name: String,
484+
alpha: Option[(Float)] = None,
485+
beta: Option[(Float)] = None,
486+
transA: Option[(Int)] = None,
487+
transB: Option[(Int)] = None,
488+
A: Option[Tensor[T]],
489+
B: Option[Tensor[T]],
490+
C: Option[Tensor[T]] = None
491+
)(
492+
implicit evT: Contains[
493+
T,
494+
Union[Float16]#or[Float]#or[Double]#or[UInt]#or[ULong]#or[Int]#or[Long]#or[UNil]#create
495+
]
496+
): (Tensor[T]) = ???
497+
424498
def Gather1[@sp T: Numeric: ClassTag, @sp Tind: Numeric: ClassTag](
425499
name: String,
426500
axis: Option[(Int)],
@@ -437,6 +511,20 @@ class NGraphBackend(onnxBytes: Array[Byte])
437511
(trinaryOpNoAttrs(name, "Gather", data, indices, None: Option[Tensor[T]]))
438512
}
439513

514+
def Gather11[@sp T: Numeric: ClassTag, @sp Tind: Numeric: ClassTag](
515+
name: String,
516+
axis: Option[(Int)] = None,
517+
data: Option[Tensor[T]],
518+
indices: Option[Tensor[Tind]]
519+
)(
520+
implicit evT: Contains[T, Union[UByte]#or[UShort]#or[UInt]#or[ULong]#or[Byte]#or[Short]#or[
521+
Int
522+
]#or[Long]#or[Float16]#or[Float]#or[Double]#or[String]#or[Boolean]#or[Complex[Float]]#or[
523+
Complex[Double]
524+
]#or[UNil]#create],
525+
evTind: Contains[Tind, Union[Int]#or[Long]#or[UNil]#create]
526+
): (Tensor[T]) = ???
527+
440528
def Conv1[@sp T: Numeric: ClassTag](
441529
name: String,
442530
auto_pad: Option[(String)] = None,
@@ -464,6 +552,19 @@ class NGraphBackend(onnxBytes: Array[Byte])
464552
trinaryOp(name, "Conv", X, W, B, map)
465553
}
466554

555+
def Conv11[@sp T: Numeric: ClassTag](
556+
name: String,
557+
auto_pad: Option[(String)] = None,
558+
dilations: Option[(Array[Int])] = None,
559+
group: Option[(Int)] = None,
560+
kernel_shape: Option[(Array[Int])] = None,
561+
pads: Option[(Array[Int])] = None,
562+
strides: Option[(Array[Int])] = None,
563+
X: Option[Tensor[T]],
564+
W: Option[Tensor[T]],
565+
B: Option[Tensor[T]] = None
566+
)(implicit evT: Contains[T, Union[Float16]#or[Float]#or[Double]#or[UNil]#create]): (Tensor[T]) = ???
567+
467568
/*
468569
val inputs: Seq[String] = node.input
469570
assert (inputs.size == 2 || inputs.size == 3, s"number of inputs of a conv node should always be 2 or 3, got ${inputs.size}")
@@ -946,6 +1047,22 @@ class NGraphBackend(onnxBytes: Array[Byte])
9461047
evI: Contains[I, Union[Long]#or[UNil]#create]
9471048
): (Tensor[T], Tensor[I]) = ???
9481049

1050+
def MaxPool11[@sp T: Numeric: ClassTag, @sp I: Numeric: ClassTag](
1051+
name: String,
1052+
auto_pad: Option[(String)] = None,
1053+
ceil_mode: Option[(Int)] = None,
1054+
dilations: Option[(Array[Int])] = None,
1055+
kernel_shape: Option[(Array[Int])],
1056+
pads: Option[(Array[Int])] = None,
1057+
storage_order: Option[(Int)] = None,
1058+
strides: Option[(Array[Int])] = None,
1059+
X: Option[Tensor[T]]
1060+
)(
1061+
implicit evT: Contains[T, Union[Float16]#or[Float]#or[Double]#or[UNil]#create],
1062+
evI: Contains[I, Union[Long]#or[UNil]#create]
1063+
): (Tensor[T], Tensor[I]) = ???
1064+
1065+
9491066
def Concat4[@sp T: Numeric: ClassTag](
9501067
name: String,
9511068
axis: Option[(Int)],
@@ -1031,6 +1148,17 @@ class NGraphBackend(onnxBytes: Array[Byte])
10311148
implicit evT: Contains[T, Union[Float16]#or[Float]#or[Double]#or[UNil]#create]
10321149
): (Tensor[T]) = ???
10331150

1151+
def AveragePool11[@sp T: Numeric: ClassTag](
1152+
name: String,
1153+
auto_pad: Option[(String)] = None,
1154+
ceil_mode: Option[(Int)] = None,
1155+
count_include_pad: Option[(Int)] = None,
1156+
kernel_shape: Option[(Array[Int])],
1157+
pads: Option[(Array[Int])] = None,
1158+
strides: Option[(Array[Int])] = None,
1159+
X: Option[Tensor[T]]
1160+
)(implicit evT: Contains[T, Union[Float16]#or[Float]#or[Double]#or[UNil]#create]): (Tensor[T]) = ???
1161+
10341162
def Reshape1[@sp T: Numeric: ClassTag](
10351163
name: String,
10361164
consumed_inputs: Option[(Array[Int])] = None,

0 commit comments

Comments
 (0)