-
Notifications
You must be signed in to change notification settings - Fork 7
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
9e8ee74
commit aec6ea5
Showing
12 changed files
with
61 additions
and
25 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
19 changes: 0 additions & 19 deletions
19
...rence/inference-core/src/commonTest/kotlin/io/kinference/operators/operations/SizeTest.kt
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
58 changes: 58 additions & 0 deletions
58
inference/inference-tfjs/src/jsMain/kotlin/io.kinference.tfjs/operators/tensor/Size.kt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,58 @@ | ||
package io.kinference.tfjs.operators.tensor | ||
|
||
import io.kinference.attribute.Attribute | ||
import io.kinference.data.ONNXData | ||
import io.kinference.graph.Contexts | ||
import io.kinference.ndarray.arrays.NDArrayTFJS | ||
import io.kinference.operator.* | ||
import io.kinference.protobuf.message.TensorProto | ||
import io.kinference.tfjs.data.tensors.TFJSTensor | ||
import io.kinference.tfjs.data.tensors.asTensor | ||
|
||
sealed class Size( | ||
name: String, | ||
info: OperatorInfo, | ||
attributes: Map<String, Attribute<Any>>, | ||
inputs: List<String>, | ||
outputs: List<String> | ||
) : Operator<TFJSTensor, TFJSTensor>(name, info, attributes, inputs, outputs) { | ||
companion object { | ||
private val DEFAULT_VERSION = VersionInfo(sinceVersion = 1) | ||
|
||
operator fun invoke(name: String, version: Int?, attributes: Map<String, Attribute<Any>>, inputs: List<String>, outputs: List<String>): Size { | ||
return when (version ?: DEFAULT_VERSION.sinceVersion) { | ||
in SizeVer1.VERSION.asRange() -> SizeVer1(name, attributes, inputs, outputs) | ||
else -> error("Unsupported version of Size operator: $version") | ||
} | ||
} | ||
} | ||
} | ||
|
||
|
||
class SizeVer1( | ||
name: String, | ||
attributes: Map<String, Attribute<Any>>, | ||
inputs: List<String>, | ||
outputs: List<String> | ||
) : Size(name, INFO, attributes, inputs, outputs) { | ||
companion object { | ||
private val ATTRIBUTES_INFO = emptyList<AttributeInfo>() | ||
|
||
private val INPUTS_INFO = listOf( | ||
IOInfo(0, PRIMITIVE_DATA_TYPES, "data", differentiable = true, optional = false) | ||
) | ||
|
||
private val OUTPUTS_INFO = listOf( | ||
IOInfo(0, setOf(TensorProto.DataType.INT64), "size", differentiable = true, optional = false) | ||
) | ||
|
||
internal val VERSION = VersionInfo(sinceVersion = 1) | ||
private val INFO = OperatorInfo("Size", ATTRIBUTES_INFO, INPUTS_INFO, OUTPUTS_INFO, VERSION, OperatorInfo.DEFAULT_DOMAIN) | ||
} | ||
|
||
override suspend fun <D : ONNXData<*, *>> apply(contexts: Contexts<D>, inputs: List<TFJSTensor?>): List<TFJSTensor?> { | ||
val inputSize = inputs[0]!!.data.linearSize | ||
val dataSize = NDArrayTFJS.intScalar(inputSize.toLong()) | ||
return listOf(dataSize.asTensor("size")) | ||
} | ||
} |
2 changes: 0 additions & 2 deletions
2
utils/utils-testing/src/commonMain/resources/size/test_size/descriptor.txt
This file was deleted.
Oops, something went wrong.
Binary file removed
BIN
-85 Bytes
utils/utils-testing/src/commonMain/resources/size/test_size/model.onnx
Binary file not shown.
1 change: 0 additions & 1 deletion
1
utils/utils-testing/src/commonMain/resources/size/test_size/test_data_set_0/input_0.pb
This file was deleted.
Oops, something went wrong.
Binary file removed
BIN
-16 Bytes
utils/utils-testing/src/commonMain/resources/size/test_size/test_data_set_0/output_0.pb
Binary file not shown.
2 changes: 0 additions & 2 deletions
2
utils/utils-testing/src/commonMain/resources/size/test_size_example/descriptor.txt
This file was deleted.
Oops, something went wrong.
Binary file removed
BIN
-89 Bytes
utils/utils-testing/src/commonMain/resources/size/test_size_example/model.onnx
Binary file not shown.
Binary file removed
BIN
-44 Bytes
.../utils-testing/src/commonMain/resources/size/test_size_example/test_data_set_0/input_0.pb
Binary file not shown.
Binary file removed
BIN
-16 Bytes
...utils-testing/src/commonMain/resources/size/test_size_example/test_data_set_0/output_0.pb
Binary file not shown.