From bc00ea04a363266ecba750424cbc2289f5e8ab48 Mon Sep 17 00:00:00 2001 From: Alex Merritt Date: Wed, 14 Oct 2020 11:53:09 -0300 Subject: [PATCH] comment --- core/src/main/scala/ONNXBytesDataSource.scala | 1 + 1 file changed, 1 insertion(+) diff --git a/core/src/main/scala/ONNXBytesDataSource.scala b/core/src/main/scala/ONNXBytesDataSource.scala index def5a944..83815213 100644 --- a/core/src/main/scala/ONNXBytesDataSource.scala +++ b/core/src/main/scala/ONNXBytesDataSource.scala @@ -8,6 +8,7 @@ class ONNXBytesDataSource(onnxBytes: Array[Byte]) extends AutoCloseable with Dat val onnxHelper = new ONNXHelper(onnxBytes) + //TODO: return non-tensor params override def getParams[T: Numeric: ClassTag](name: String): Tensor[T] = { val params = onnxHelper.params.filter(x => x._1 == name).headOption params match {