We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d2c82b3 commit bc00ea0Copy full SHA for bc00ea0
core/src/main/scala/ONNXBytesDataSource.scala
@@ -8,6 +8,7 @@ class ONNXBytesDataSource(onnxBytes: Array[Byte]) extends AutoCloseable with Dat
8
9
val onnxHelper = new ONNXHelper(onnxBytes)
10
11
+ //TODO: return non-tensor params
12
override def getParams[T: Numeric: ClassTag](name: String): Tensor[T] = {
13
val params = onnxHelper.params.filter(x => x._1 == name).headOption
14
params match {
0 commit comments