Skip to content

Commit 92b4a19

Browse files
committed
Fix Shape op constraint
1 parent 9acb5f3 commit 92b4a19

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

core/src/main/scala/ONNX.scala

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -933,12 +933,10 @@ package object onnx {
933933
(callOp(name, "Selu", allInputs, map))
934934
}
935935
}
936-
//Missing in NDScala - ready - P1
936+
937937
trait ShapeV1 extends Operator {
938938
def ShapeV1[
939-
@sp T <: UByte | UShort | UInt | ULong | Byte | Short | Int | Long | Float16 | Float | Double | String | Boolean | Complex[
940-
Float
941-
] | Complex[Double]: Numeric,
939+
@sp T <: UByte | UShort | UInt | ULong | Byte | Short | Int | Long | Float16 | Float | Double | Boolean | String | Complex[Float] | Complex[Double],
942940
@sp T1 <: Long: Numeric
943941
, Tt <: TensorTypeDenotation, Td <: TensorShapeDenotation, S <: Shape, Tt1 <: TensorTypeDenotation, Td1 <: TensorShapeDenotation](name: String, data: Tensor[T, Tuple3[Tt,Td,S]])(using tt: ValueOf[Tt1], td: TensorShapeDenotationOf[Td1], s: ShapeOf[io.kjaer.compiletime.Shape.Rank[S] & Dimension #: SNil]): Tensor[T1, Tuple3[Tt1,Td1,io.kjaer.compiletime.Shape.Rank[S] & Dimension #: SNil]] = {
944942
val map: Map[String, Any] = Map()

0 commit comments

Comments
 (0)